Come creare un grafico a dispersione con una linea di regressione in r
Spesso quando eseguiamo una semplice regressione lineare, vogliamo creare un grafico a dispersione per visualizzare le diverse combinazioni dei valori xey.
Fortunatamente, R semplifica la creazione di nuvole di punti utilizzando la funzione plot() . Per esempio:
#create some fake data data <- data.frame(x = c(1, 1, 2, 3, 4, 4, 5, 6, 7, 7, 8, 9, 10, 11, 11), y = c(13, 14, 17, 12, 23, 24, 25, 25, 24, 28, 32, 33, 35, 40, 41)) #create scatterplot of data plot(data$x, data$y)
È anche semplice aggiungere una linea di regressione al grafico a dispersione utilizzando la funzione abline() .
Per esempio:
#fit a simple linear regression model model <- lm(y ~ x, data = data) #add the fitted regression line to the scatterplot abline(model)
Possiamo anche aggiungere linee di intervallo di confidenza al grafico utilizzando la funzione predit() :
#define range of x values newx = seq(min(data$x),max(data$x),by = 1) #find 95% confidence interval for the range of x values conf_interval <- predict(model, newdata=data.frame(x=newx), interval="confidence", level = 0.95) #create scatterplot of values with regression line plot(data$x, data$y) abline(model) #add dashed lines (lty=2) for the 95% confidence interval lines(newx, conf_interval[,2], col="blue", lty=2) lines(newx, conf_interval[,3], col="blue", lty=2)
Oppure potremmo invece aggiungere linee di intervallo di previsione al grafico specificando il tipo di intervallo nella funzione predit() :
#define range of x values newx = seq(min(data$x),max(data$x),by = 1) #find 95% prediction interval for the range of x values pred_interval <- predict(model, newdata=data.frame(x=newx), interval="prediction" , level = 0.95) #create scatterplot of values with regression line plot(data$x, data$y) abline(model) #add dashed lines (lty=2) for the 95% confidence interval lines(newx, pred_interval[,2], col="red", lty=2) lines(newx, pred_interval[,3], col="red", lty=2)
Infine, possiamo rendere la trama esteticamente più gradevole aggiungendo un titolo, modificando i nomi degli assi e modificando la forma dei singoli punti della trama.
plot(data$x, data$y, main = "Scatterplot of x vs. y", #add title pch=16, #specify points to be filled in xlab='x', #change x-axis name ylab='y') #change y-axis name abline(model, col='steelblue') #specify color of regression line
Risorse addizionali
I seguenti tutorial spiegano come eseguire altre attività comuni in R:
Come etichettare i punti su un grafico a dispersione in R
Come utilizzare la funzione Jitter in R per le nuvole di punti