如何在ggplot中实现fill参数动态切换：单颜色或列映射？

完全可以让参数z同时支持裸列名和字符串输入，用rlang包的ensym()函数就能实现——它会自动识别传入的是裸变量名还是字符串，统一转换成ggplot能够解析的符号对象。

下面是整合了单一颜色、分组颜色两种场景的简化函数，既支持z参数传入裸列名或字符串，又避免了冗余的if-else分支：

library(ggplot2)
library(dplyr)
library(RColorBrewer)
library(rlang)

f1 <- function(data, x, y, z = NULL, color = "red") {
  # 按x变量计数（和原逻辑保持一致）
  data <- data %>% count({{x}})
  
  # 动态生成fill参数：有z则按z分组配色，无z则用单一颜色
  fill_aes <- if (!is.null(enexpr(z))) {
    expr(factor(!!ensym(z)))
  } else {
    color
  }
  
  # 构建ggplot对象，仅当z存在时添加配色刻度
  ggplot(data, aes(x = factor({{x}}), y = {{y}})) +
    geom_col(width = 0.6, fill = fill_aes) +
    theme_minimal() +
    (if (!is.null(enexpr(z))) scale_fill_brewer(palette = "Set1") else NULL)
}

• 单一颜色模式（不传z参数）：

f1(mtcars, cyl, n, color = "blue")

• 分组配色模式（传裸列名）：

f1(mtcars, cyl, n, z = cyl)

• 分组配色模式（传字符串列名）：

f1(mtcars, cyl, n, z = "cyl")

• ensym(z)：把传入的裸变量（如cyl）或字符串（如"cyl"）统一转换为符号cyl，确保ggplot能正确识别为数据框的列。
• enexpr(z)：判断z参数是否被传入（避免NULL值干扰），以此动态决定是否启用分组配色。
• 用(if (...) ... else NULL)的方式向ggplot添加刻度，这种写法能在条件不满足时自动忽略该层，比单独分分支更简洁。

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