循环中赋值ggplot对象异常：多项式回归绘图全为4阶问题求助

你遇到的问题是在循环中生成1-4阶多项式回归图，但所有子图最终都显示4阶拟合结果，这是ggplot延迟求值特性导致的。

问题代码回顾 #

library(ggplot2)
library(gridExtra)

set.seed(123)
x <- 1:10
y <- sin(2*pi*x) + 0.2*rnorm(x)
df <- data.frame(x, y)

for (i in 1:4) {
  pp <- ggplot(df, aes(x, y)) +
    geom_point() +
    stat_smooth(method='lm', formula = y ~ poly(x, i))
  nam <- paste0("p", i)
  assign(nam, pp)
}

grid.arrange(p1, p2, p3, p4, nrow=2, ncol=2)

问题原因 #

ggplot对象不会立即计算表达式中的变量值，而是在实际渲染图形时才去读取变量的当前值。循环结束后，变量i的最终值是4，所以所有生成的ggplot对象在渲染时都会使用i=4，导致所有图都是4阶拟合。

修复方案 #

方案1：使用局部环境锁定变量值

在循环内创建局部环境，把当前的i值固定下来，避免后续循环修改影响已生成的ggplot对象：

library(ggplot2)
library(gridExtra)

set.seed(123)
x <- 1:10
y <- sin(2*pi*x) + 0.2*rnorm(x)
df <- data.frame(x, y)

plot_list <- list()
for (i in 1:4) {
  # 创建局部环境，锁定当前i的值
  local({
    i_local <- i
    p <- ggplot(df, aes(x, y)) +
      geom_point() +
      stat_smooth(method='lm', formula = y ~ poly(x, i_local)) +
      ggtitle(paste0(i_local, "阶多项式拟合"))
    plot_list[[i_local]] <<- p
  })
}

grid.arrange(grobs = plot_list, nrow=2, ncol=2)

方案2：使用lapply替代循环（更推荐）

用lapply代替for循环，每个迭代都会创建独立的作用域，自动锁定当前的阶数值，代码更简洁：

library(ggplot2)
library(gridExtra)

set.seed(123)
x <- 1:10
y <- sin(2*pi*x) + 0.2*rnorm(x)
df <- data.frame(x, y)

plot_list <- lapply(1:4, function(i) {
  ggplot(df, aes(x, y)) +
    geom_point() +
    stat_smooth(method='lm', formula = y ~ poly(x, i)) +
    ggtitle(paste0(i, "阶多项式拟合"))
})

grid.arrange(grobs = plot_list, nrow=2, ncol=2)

说明 #

两种方案都能解决延迟求值的问题，其中lapply的写法更符合R的函数式编程风格，代码可读性和维护性更好，同时避免了使用assign这类容易引发问题的函数。

内容的提问来源于stack exchange，提问作者JACKY88