如何以ggplot的y变量为条件用case_when动态设置面板背景色

完整实现代码 #

library(ggplot2)
library(dplyr)

# 基础数据准备
BLUEBAND <- c(1,2,4,8,16,32,64)
GREENBAND <- c(1,3,9,27,81,243,729)
REDBAND <- c(1,2,3,4,5,6,7)
LEVEL99 <- c(1,2,3,4,5,6,7)
BGR_DF <- as.data.frame(cbind(BLUEBAND,GREENBAND,REDBAND,LEVEL99))

# 修正原代码颜色定义bug：十六进制颜色需加引号
BGRCOLORS <- c("#E6F1FF", "#ECFFE6", "#FFEBE6")

# 封装绘图函数，自动识别y变量匹配背景色
band_plot <- function(data, y_var) {
  # 捕获传入的y变量，提取字符串名称
  y_name <- as.character(substitute(y_var))
  
  ggplot(data, aes(x = LEVEL99, y = {{y_var}})) +
    geom_point() +
    geom_smooth(method = lm) +
    theme(panel.background = element_rect(fill = case_when(
      y_name == "BLUEBAND" ~ BGRCOLORS[1],
      y_name == "GREENBAND" ~ BGRCOLORS[2],
      y_name == "REDBAND" ~ BGRCOLORS[3],
      # 匹配所有其他情况，返回灰色
      TRUE ~ "#F0F0F0"
    )))
}

使用示例 #

调用时仅需指定y轴变量，背景色会自动匹配：

• 绘制绿波段自动匹配绿色背景：band_plot(BGR_DF, GREENBAND)
• 绘制红波段自动匹配红色背景：band_plot(BGR_DF, REDBAND)
• 绘制非三类波段的变量自动匹配灰色背景：band_plot(BGR_DF, LEVEL99)

问题解答 #

核心问题：获取ggplot的y变量作为判断条件 #

将绘图逻辑封装为自定义函数，通过substitute()捕获传入的y变量表达式，转为字符串即可直接拿到变量名，无需额外绑定全局变量做判断。其中{{y_var}}是rlang的curly-curly语法，用于在函数内正确传递ggplot的美学映射变量。

附加问题：case_when默认匹配规则 #

case_when()支持将TRUE ~ <默认值>作为最后一个判断条件，所有未被前面条件匹配的场景都会触发该分支，直接设置为需要的默认颜色即可。

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