R语言ggplot2分组时序体积图：列表模式配色问题排查

问题背景 #

用R的ggplot2编写了一个绘制分组变量随时间变化体积的函数，设置了plot_type参数，可选"facet"（分面）或"list"（列表）。分面图显示效果符合预期，但列表模式下的子图没有沿用分面图的分组配色（甚至无配色），推测是代码执行顺序问题导致配色未传递，需要排查。

原函数代码 #

plot_group_vot <-  function(data = data,
                            group_var = group_var,
                            date_var = date,
                            unit = c("day", "week", "month", "quarter", "year"),
                            nrow = 2,
                            plot_type = c("facet", "list")){

  unit <- match.arg(unit)

  date_sym <- rlang::ensym(date_var)
  group_sym <- rlang::ensym(group_var)


  if (plot_type == "facet") {
    data <- data %>% dplyr::mutate(plot_date = lubridate::floor_date(!!date_sym, unit = unit),
                                   facet_var = !!group_sym)

    plot <- data %>%
      dplyr::count(plot_date, facet_var) %>%
      ggplot2::ggplot(ggplot2::aes(x = plot_date, y = n, fill = facet_var)) +
      ggplot2::geom_col() +
      ggplot2::theme_minimal() +
      ggplot2::scale_x_date(date_breaks = "1 months", date_labels = "%d-%b") +
      ggplot2::theme(legend.position = "none",
                     axis.text.x = element_text(angle = 45),
                     panel.grid = element_blank()) +
      ggplot2::labs(title = "Topic Volume over Time", x = NULL, y = "Number of Posts") +
      ggplot2::facet_wrap(~facet_var, nrow = nrow)
    print(plot)

  } else if (plot_type == "list") {
    data <- data %>% dplyr::mutate(plot_date = lubridate::floor_date(!!date_sym, unit = unit),
                                   facet_var = !!group_sym)

    unique_groups <- data %>% pull(!!group_sym) %>% unique()
    plots <- list()

    for (group in unique_groups) {
      plot <- data %>%
        dplyr::filter(!!group_sym == group) %>%
        dplyr::count(plot_date) %>%
        ggplot2::ggplot(ggplot2::aes(x = plot_date, y = n)) +
        ggplot2::geom_col() +
        ggplot2::theme_minimal() +
        ggplot2::scale_x_date(date_breaks = "1 months", date_labels = "%d-%b") +
        ggplot2::theme(legend.position = "none",
                       axis.text.x = element_text(angle = 45),
                       panel.grid = element_blank()) +
        ggplot2::labs(title = paste("Topic Volume over Time -", group), x = NULL, y = "Number of Posts")
      plots[[group]] <- plot
    }
    return(plots)

  } else {
    stop("Invalid plot_type argument. Try 'facet' or 'list'.")
  }
}

调用方式 #

plot_group_vot(data = data, group_var = group_var, date_var = date, unit = "day", plot_type = "list")

问题原因 #

1. 列表模式未设置fill映射：分面模式的ggplot(aes)里指定了fill = facet_var，基于全部分组生成统一配色；但列表模式的ggplot(aes)里没有fill参数，所以柱子无填充色。
2. 单分组数据无法生成配色映射：列表模式中每个子图只筛选了单个分组的数据，即便加上fill，ggplot也只会识别到一个分组值，无法自动生成和分面图一致的多分组配色体系。

解决方案 #

核心是先基于全部分组定义统一的配色方案，再在每个子图中应用该配色：

1. 在列表模式代码块开头，提取所有分组并生成对应的配色（用ggplot默认的离散配色即可）；
2. 在子图的ggplot(aes)中添加fill参数，绑定当前分组；
3. 用scale_fill_manual指定全局配色，确保每个子图的颜色和分面图对应分组一致。

修改后的完整函数代码 #

plot_group_vot <-  function(data = data,
                            group_var = group_var,
                            date_var = date,
                            unit = c("day", "week", "month", "quarter", "year"),
                            nrow = 2,
                            plot_type = c("facet", "list")){

  unit <- match.arg(unit)

  date_sym <- rlang::ensym(date_var)
  group_sym <- rlang::ensym(group_var)


  if (plot_type == "facet") {
    data <- data %>% dplyr::mutate(plot_date = lubridate::floor_date(!!date_sym, unit = unit),
                                   facet_var = !!group_sym)

    plot <- data %>%
      dplyr::count(plot_date, facet_var) %>%
      ggplot2::ggplot(ggplot2::aes(x = plot_date, y = n, fill = facet_var)) +
      ggplot2::geom_col() +
      ggplot2::theme_minimal() +
      ggplot2::scale_x_date(date_breaks = "1 months", date_labels = "%d-%b") +
      ggplot2::theme(legend.position = "none",
                     axis.text.x = element_text(angle = 45),
                     panel.grid = element_blank()) +
      ggplot2::labs(title = "Topic Volume over Time", x = NULL, y = "Number of Posts") +
      ggplot2::facet_wrap(~facet_var, nrow = nrow)
    print(plot)

  } else if (plot_type == "list") {
    data <- data %>% dplyr::mutate(plot_date = lubridate::floor_date(!!date_sym, unit = unit),
                                   facet_var = !!group_sym)

    unique_groups <- data %>% pull(!!group_sym) %>% unique()
    # 生成与分面图一致的配色：用ggplot默认离散配色
    group_colors <- ggplot2::scale_fill_discrete()$palette(length(unique_groups))
    names(group_colors) <- unique_groups
    
    plots <- list()

    for (group in unique_groups) {
      plot <- data %>%
        dplyr::filter(!!group_sym == group) %>%
        dplyr::count(plot_date) %>%
        ggplot2::ggplot(ggplot2::aes(x = plot_date, y = n, fill = !!group_sym)) +
        ggplot2::geom_col() +
        ggplot2::theme_minimal() +
        ggplot2::scale_x_date(date_breaks = "1 months", date_labels = "%d-%b") +
        ggplot2::scale_fill_manual(values = group_colors) +  # 应用统一配色
        ggplot2::theme(legend.position = "none",
                       axis.text.x = element_text(angle = 45),
                       panel.grid = element_blank()) +
        ggplot2::labs(title = paste("Topic Volume over Time -", group), x = NULL, y = "Number of Posts")
      plots[[group]] <- plot
    }
    return(plots)

  } else {
    stop("Invalid plot_type argument. Try 'facet' or 'list'.")
  }
}

关键修改点 #

• 新增group_colors：基于所有分组生成和分面图一致的配色方案；
• 子图aes中添加fill = !!group_sym：为柱子设置填充映射；
• 添加scale_fill_manual(values = group_colors)：强制每个子图使用统一的分组配色，确保和分面图对应分组颜色一致。

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