如何在R函数的ggplot2 geom_bar中按Y值条件设置条形填充色

需求：基于给定数据集DATA编写my_function函数，实现以下效果：

• 绘制分组（按Year分面）的条形图，Y轴为每个面板内的计数占比（当前条形计数/面板总计数）
• 当占比小于0.05或大于0.95时，条形设为白色（隐藏显示）
• 其余情况按X变量的值填充颜色

修正后的实现代码 #

library(tidyverse)
library(rlang)

DATA <- structure(list(Year = c(2015, 2015, 2015, 2015, 2015, 2016, 2016, 
2016, 2016, 2016), X = c("Dissatisfied", "Dissatisfied", 
"Dissatisfied", "Dissatisfied", "Dissatisfied", "Satisfied", 
"Dissatisfied", "Dissatisfied", "Satisfied", "Dissatisfied")), class = "data.frame", row.names = c(NA, 
-10L))

my_function <- function(data = DATA, x = X, cols = vars(Year), fill = X){
  ggplot(data) +
    aes(
      x = {{x}},
      y = after_stat(count / sum(count)),  # 简化面板内占比计算
      fill = ifelse(
        after_stat(count / sum(count)) < 0.05 | after_stat(count / sum(count)) > 0.95,
        "white",  # 满足阈值条件设为白色
        as.character({{fill}})  # 否则使用指定变量的实际值填充
      )
    ) +
    geom_bar(stat = "count") +
    facet_grid(cols = cols) +
    theme(legend.title = element_blank()) +
    # 可选：自定义X变量类别的对应颜色，确保图例仅显示有效类别
    scale_fill_manual(
      values = c("Dissatisfied" = "#1f77b4", "Satisfied" = "#ff7f0e"),
      breaks = c("Dissatisfied", "Satisfied")
    )
}

# 调用示例
my_function()

关键修改说明 #

1. 简化占比计算：用after_stat(count / sum(count))替代原代码的sapply写法，ggplot会自动在每个分面面板内计算总计数，逻辑更简洁且符合ggplot统计语法规范。
2. 正确映射填充颜色：
   • 使用{{fill}}引用函数参数中的填充变量（遵循rlang的tidy eval语法）
   • 用as.character()将因子类型的X变量转为字符，避免颜色映射时的类型冲突
   • 当占比超出阈值时返回固定值"white"，否则返回fill变量的实际类别值
3. 图例优化：通过scale_fill_manual明确指定X变量类别的颜色，并设置breaks确保图例只显示有效类别（白色不会出现在图例中）

效果验证 #

• 2015年面板中，Dissatisfied占比为1（>0.95），条形显示为白色
• 2016年面板中，Dissatisfied占比0.8、Satisfied占比0.2（均在0.05-0.95区间），分别显示对应颜色

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