R语言ggplot循环添加多折线时着色异常问题求助

我想通过循环在ggplot中添加不同折线，并为每条折线设置对应颜色，尝试了如下代码：

scenario_colors <- c(
  "scenario1" = "black",
  "scenario2" = "red",
  "scenario3" = "green"
)
all_scenario_names <- c("scenario1", "scenario2", "scenario3")
p <- p + scale_color_manual(values = scenario_colors)

for (j in 1:3){
  column <- column_names[j]
  scenario <- all_scenario_names[j]
  print(paste0(column, " ---- ", scenario, " ---- ", scenario_colors[scenario]))
  p <- p + geom_line(data=rama_data, aes(y = .data[[column]], group = 1, color=scenario))
}

但最终绘图异常，所有3条折线都显示为scenario3对应的绿色。

而采用硬编码的方式，每条折线能正常显示对应颜色（黑色、红色、绿色）：

for (j in 1:3){
  column <- column_names[j]
  scenario <- all_scenario_names[j]
  print(paste0(column, " ---- ", scenario, " ---- ", scenario_colors[scenario]))
  if (j == 1){p <- p + geom_line(data=rama_data, aes(y = .data[[column]], group = 1, color="scenario1"))}
  else if(j == 2){p <- p + geom_line(data=rama_data, aes(y = .data[[column]], group = 1, color="scenario2"))}
  else{p <- p + geom_line(data=rama_data, aes(y = .data[[column]], group = 1, color="scenario3"))}
}

硬编码方式无法适配未来扩展的场景数量，请问第一段代码的问题出在哪里？

补充使用的数据结构：

variable1 <- c(runif(5))
variable2 <- c(3*runif(5))
variable3 <- c(5 + runif(5))
period <- c("MM012021", "MM022021", "MM032021", "MM042021", "MM052021")
# Create the data.table
column_names <- c("variable1", "variable2", "variable3")
rama_data <- data.table(period, variable1, variable2, variable3)

这是ggplot的延迟求值机制导致的：在循环中使用aes(color=scenario)时，scenario不会被立即解析为当前循环迭代的字符串值，而是保存对变量scenario的引用。当最终绘制图形时，循环已经执行完毕，scenario的值停留在最后一次迭代的scenario3，所以所有折线都使用了这个值对应的颜色。

而硬编码的color="scenario1"是直接传入字符串常量，不存在引用延迟的问题，因此能正常显示对应颜色。

方法1：修正循环中的aes赋值 #

在循环中，需要把scenario的当前值强制解析为字符串常量传入aes，可以用!!运算符实现：

scenario_colors <- c(
  "scenario1" = "black",
  "scenario2" = "red",
  "scenario3" = "green"
)
all_scenario_names <- c("scenario1", "scenario2", "scenario3")
# 初始化ggplot对象
p <- ggplot(rama_data, aes(x = period)) + 
  scale_color_manual(values = scenario_colors)

for (j in 1:3){
  column <- column_names[j]
  scenario <- all_scenario_names[j]
  # 使用!!强制解析当前scenario的字符串值
  p <- p + geom_line(aes(y = .data[[column]], group = 1, color = !!scenario))
}
# 绘制图形
p

方法2：转换为长数据格式（推荐） #

ggplot的设计更适配长数据（每个观测占一行），而非当前的宽数据格式，这种方式无需循环，代码更简洁且易于扩展：

步骤1：转换数据格式 #

library(data.table)
# 将宽数据转换为长数据
rama_long <- melt(rama_data, 
                  id.vars = "period", 
                  variable.name = "scenario", 
                  value.name = "value")
# 替换变量名为对应的场景名称
rama_long[, scenario := factor(scenario, 
                               levels = column_names, 
                               labels = all_scenario_names)]

步骤2：绘制图形 #

ggplot(rama_long, aes(x = period, y = value, color = scenario, group = scenario)) +
  geom_line() +
  scale_color_manual(values = scenario_colors)

未来新增场景时，只需更新column_names、all_scenario_names和scenario_colors即可，完全适配扩展需求。

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