1. ホーム
  2. r

[解決済み】回帰線の式とR^2をグラフに追加する

2022-03-31 14:05:53

質問

回帰直線の式とR^2を追加する方法を教えてください。 ggplot . 私のコードは、次のとおりです。

library(ggplot2)

df <- data.frame(x = c(1:100))
df$y <- 2 + 3 * df$x + rnorm(100, sd = 40)
p <- ggplot(data = df, aes(x = x, y = y)) +
            geom_smooth(method = "lm", se=FALSE, color="black", formula = y ~ x) +
            geom_point()
p

何かお手伝いいただけることがあれば、ぜひお願いします。

解決方法は?

以下はその解決策の一つです。

# GET EQUATION AND R-SQUARED AS STRING
# SOURCE: https://groups.google.com/forum/#!topic/ggplot2/1TgH-kG5XMA

lm_eqn <- function(df){
    m <- lm(y ~ x, df);
    eq <- substitute(italic(y) == a + b %.% italic(x)*","~~italic(r)^2~"="~r2, 
         list(a = format(unname(coef(m)[1]), digits = 2),
              b = format(unname(coef(m)[2]), digits = 2),
             r2 = format(summary(m)$r.squared, digits = 3)))
    as.character(as.expression(eq));
}

p1 <- p + geom_text(x = 25, y = 300, label = lm_eqn(df), parse = TRUE)

EDIT このコードを拾ったソースがわかりました。以下は リンク ggplot2 の google グループにある元の投稿にアクセスします。