276

に回帰直線方程式と 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

どんな助けでも大歓迎です。

4

9 に答える 9

280

ここに1つの解決策があります

# 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)

編集。このコードを選んだソースを見つけました。ggplot2 google グループの元の投稿へのリンクは次のとおりです。

出力

于 2011-09-26T01:20:09.703 に答える
104

および関連する関数のソースの数行を変更してstat_smooth、近似式と R 二乗値を追加する新しい関数を作成しました。これはファセット プロットでも機能します。

library(devtools)
source_gist("524eade46135f6348140")
df = data.frame(x = c(1:100))
df$y = 2 + 5 * df$x + rnorm(100, sd = 40)
df$class = rep(1:2,50)
ggplot(data = df, aes(x = x, y = y, label=y)) +
  stat_smooth_func(geom="text",method="lm",hjust=0,parse=TRUE) +
  geom_smooth(method="lm",se=FALSE) +
  geom_point() + facet_wrap(~class)

ここに画像の説明を入力

@Ramnathの回答のコードを使用して、方程式をフォーマットしました。このstat_smooth_func関数はあまり堅牢ではありませんが、いじってみるのは難しくありません。

https://gist.github.com/kdauria/524eade46135f6348140 . ggplot2エラーが出たら更新してみてください。

于 2015-01-15T08:34:57.807 に答える
79

Ramnath の投稿を変更して、a) より一般的なものにして、データ フレームではなく線形モデルをパラメーターとして受け入れるようにし、b) ネガをより適切に表示するようにしました。

lm_eqn = function(m) {

  l <- list(a = format(coef(m)[1], digits = 2),
      b = format(abs(coef(m)[2]), digits = 2),
      r2 = format(summary(m)$r.squared, digits = 3));

  if (coef(m)[2] >= 0)  {
    eq <- substitute(italic(y) == a + b %.% italic(x)*","~~italic(r)^2~"="~r2,l)
  } else {
    eq <- substitute(italic(y) == a - b %.% italic(x)*","~~italic(r)^2~"="~r2,l)    
  }

  as.character(as.expression(eq));                 
}

使用方法は次のように変更されます。

p1 = p + geom_text(aes(x = 25, y = 300, label = lm_eqn(lm(y ~ x, df))), parse = TRUE)
于 2012-11-19T10:09:19.630 に答える
4

dplyr別のオプションは、 とbroomライブラリを使用して方程式を生成するカスタム関数を作成することです。

get_formula <- function(model) {
  
  broom::tidy(model)[, 1:2] %>%
    mutate(sign = ifelse(sign(estimate) == 1, ' + ', ' - ')) %>% #coeff signs
    mutate_if(is.numeric, ~ abs(round(., 2))) %>% #for improving formatting
    mutate(a = ifelse(term == '(Intercept)', paste0('y ~ ', estimate), paste0(sign, estimate, ' * ', term))) %>%
    summarise(formula = paste(a, collapse = '')) %>%
    as.character
  
}

lm(y ~ x, data = df) -> model
get_formula(model)
#"y ~ 6.22 + 3.16 * x"

scales::percent(summary(model)$r.squared, accuracy = 0.01) -> r_squared

次に、プロットにテキストを追加する必要があります。

p + 
  geom_text(x = 20, y = 300,
            label = get_formula(model),
            color = 'red') +
  geom_text(x = 20, y = 285,
            label = r_squared,
            color = 'blue')

プロット

于 2020-10-15T16:55:33.337 に答える