Como adicionar uma equação de regressão a um gráfico em r

Muitas vezes você pode querer adicionar uma equação de regressão a um gráfico em R da seguinte maneira:

Felizmente, isso é muito fácil de fazer usando as funções dos pacotes ggplot2 e ggpubr .

Este tutorial fornece um exemplo passo a passo de como usar funções nesses pacotes para adicionar uma equação de regressão a um gráfico em R.

Etapa 1: crie os dados

Primeiro, vamos criar alguns dados falsos para trabalhar:

 #make this example reproducible
set. seeds (1)

#create data frame
df <- data. frame (x = c(1:100))
df$y <- 4*df$x + rnorm(100, sd=20)

#view head of data frame
head(df)

  xy
1 1 -8.529076
2 2 11.672866
3 3 -4.712572
4 4 47.905616
5 5 26.590155
6 6 7.590632

Passo 2: Crie o gráfico com a equação de regressão

A seguir, usaremos a seguinte sintaxe para criar um gráfico de dispersão com uma linha de regressão ajustada e uma equação:

 #load necessary libraries
library (ggplot2)
library (ggpubr)

#create plot with regression line and regression equation
ggplot(data=df, aes (x=x, y=y)) +
        geom_smooth(method=" lm ") +
        geom_point() +
        stat_regline_equation(label. x =30, label. y =310) 

Isso nos diz que a equação de regressão ajustada é:

y = 2,6 + 4*(x)

Observe que label.x e label.y especificam as coordenadas (x,y) da equação de regressão a ser exibida.

Etapa 3: adicione R-quadrado ao gráfico (opcional)

Você também pode adicionar o valor R ao quadrado do modelo de regressão se quiser usar a seguinte sintaxe:

 #load necessary libraries
library (ggplot2)
library (ggpubr)

#create plot with regression line, regression equation, and R-squared
ggplot(data=df, aes (x=x, y=y)) +
        geom_smooth(method=" lm ") +
        geom_point() +
        stat_regline_equation(label. x =30, label. y =310) +
stat_cor( aes (label=..rr.label..), label. x =30, label. y =290) 

O R ao quadrado para este modelo é 0,98 .

Você pode encontrar mais tutoriais de R nesta página .