繁体   English   中英

如何修复图形上回归线方程的标签(使用ggplot2)

[英]How to fix the labeling of a regression line equation on graph (using ggplot2)

当我在rstudio中运行以下代码(请参见下文)时,除线性回归线方程式之外,其他所有内容看起来都很不错。 我没有得到y = 3.142x -4.751,而是得到y = c(3.142)xc(-4.751)。

我该怎么做才能解决此问题? 提前非常感谢您。

set.seed(3L)
library(ggplot2)

df <- data.frame(x = c(1:100))
df$y <- 2 + 3 * df$x + rnorm(100, sd = 40)
lm_eqn <- function(df){

# browser()
m <- lm(y ~ x, df)
a <- coef(m)[1]
a <- ifelse(sign(a) >= 0, 
     paste0(" + ", format(a, digits = 4)), 
     paste0(" - ", format(-a, digits = 4))  )

eq1 <- substitute( paste( italic(y) == b, italic(x), a ),
     list(a = a, 
          b = format(coef(m)[2], digits=4)))

eq2 <- substitute( paste( italic(R)^2 == r2 ), 
     list(r2 = format(summary(m)$r.squared, digits = 3)))

c( as.character(as.expression(eq1)), as.character(as.expression(eq2)))
}

labels <- lm_eqn(df)

p <- ggplot(data = df, aes(x = x, y = y)) +
geom_smooth(method = "lm", se=FALSE, color="red", formula = y ~ x) +
geom_point() +
geom_text(x = 75, y = 90, label = labels[1], parse = TRUE,  check_overlap = TRUE ) +
geom_text(x = 75, y = 70, label = labels[2], parse = TRUE, check_overlap = TRUE )

print(p)

在函数lm_eqnformat函数给出了命名向量,通过删除名称可以解决问题,如下所示:

set.seed(3L)
library(ggplot2)

df <- data.frame(x = c(1:100))
df$y <- 2 + 3 * df$x + rnorm(100, sd = 40)
lm_eqn <- function(df){

  # browser()
  m <- lm(y ~ x, df)
  a <- coef(m)[1]
  a <- ifelse(sign(a) >= 0, 
              paste0(" + ", format(a, digits = 4)), 
              paste0(" - ", format(-a, digits = 4))  )

  b <-  format(coef(m)[2], digits=4)
  names(a) <- names(b) <-NULL
  eq1 <- substitute( paste( italic(y) == b, italic(x), a ),
                     list(a = a, 
                          b = b))

  eq2 <- substitute( paste( italic(R)^2 == r2 ), 
                     list(r2 = format(summary(m)$r.squared, digits = 3)))

  c( as.character(as.expression(eq1)), as.character(as.expression(eq2)))
}

labels <- lm_eqn(df)

p <- ggplot(data = df, aes(x = x, y = y)) +
  geom_smooth(method = "lm", se=FALSE, color="red", formula = y ~ x) +
  geom_point() +
  geom_text(x = 75, y = 90, label = labels[1], parse = TRUE,  check_overlap = TRUE ) +
  geom_text(x = 75, y = 70, label = labels[2], parse = TRUE, check_overlap = TRUE )

print(p)

结果:

在此处输入图片说明

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM