如何在生成ggplot2折线图后修改并统一颜色以合并图例

当使用patchwork拼接多个ggplot2折线图时，若各图表的颜色分组不统一，会生成多个独立图例。手动指定颜色映射虽能合并图例，但需提前知晓所有分组，灵活性不足。我们可以仅通过已有的ggplot对象，自动提取所有分组信息并生成统一颜色映射，实现图例的自动合并。

原始示例（直接拼接生成多图例） #

library("tibble")
library("ggplot2")
library("patchwork")
 
d1 <- 
  tribble(
    ~g,  ~x, ~y,
    "a",  1,  1,
    "a",  2,  2,
    "b",  1,  2,
    "b",  2,  1
    )

p1 <- ggplot(d1, aes(x = x, y = y, group = g)) + geom_line(aes(color = g))

d2 <- 
  tribble(
    ~g,  ~x, ~y,
    "b",  1,  3,
    "b",  2,  5,
    "c",  1,  5,
    "c",  2,  4
    )

p2 <- ggplot(d2, aes(x = x, y = y, group = g)) + geom_line(aes(color = g))

# 直接拼接会生成两个独立图例
p1 + p2 + plot_layout(guides = "collect") 

手动指定颜色映射（可行但非自动化） #

# 手动定义所有分组的颜色映射
cols <- c("a" = "red", "b" = "blue", "c" = "green")

p1a <- p1 + scale_colour_manual(values = cols)
p2a <- p2 + scale_colour_manual(values = cols)

# 拼接后合并为单个图例
p1a + p2a + plot_layout(guides = "collect") 

自动化解决方案 #

实现思路 #

1. 从每个ggplot对象中提取颜色映射对应的分组变量的所有取值
2. 收集所有唯一分组值，生成统一的颜色映射（可使用默认调色板或自定义调色板）
3. 批量为所有图表应用统一的颜色映射

代码实现 #

library("tibble")
library("ggplot2")
library("patchwork")
library("purrr") # 用于批量处理图表

# 原始数据与图表
d1 <- tribble(
  ~g,  ~x, ~y,
  "a", 1, 1, "a", 2, 2, "b", 1, 2, "b", 2, 1
)
p1 <- ggplot(d1, aes(x = x, y = y, group = g)) + geom_line(aes(color = g))

d2 <- tribble(
  ~g,  ~x, ~y,
  "b", 1, 3, "b", 2, 5, "c", 1, 5, "c", 2, 4
)
p2 <- ggplot(d2, aes(x = x, y = y, group = g)) + geom_line(aes(color = g))

# 1. 提取单个图表的颜色分组值
get_color_groups <- function(plot_obj) {
  # 获取图层中颜色映射的变量名
  color_var <- as.character(plot_obj$layers[[1]]$aes_params$color)
  # 提取该变量在对应数据中的所有唯一值
  unique(plot_obj$data[[color_var]])
}

# 收集所有图表的分组值并去重
all_groups <- unique(c(get_color_groups(p1), get_color_groups(p2)))

# 2. 生成统一颜色映射（使用ggplot2默认调色板）
default_colors <- scales::hue_pal()(length(all_groups))
names(default_colors) <- all_groups

# 3. 批量更新所有图表的颜色映射
update_plot_colors <- function(plot_obj, color_scale) {
  plot_obj + scale_colour_manual(values = color_scale)
}

plots_updated <- map(list(p1, p2), update_plot_colors, color_scale = default_colors)

# 4. 拼接图表并合并图例
reduce(plots_updated, `+`) + plot_layout(guides = "collect")

说明 #

• get_color_groups函数自动从ggplot对象中提取颜色映射对应的分组变量，以及该变量的所有取值
• 可替换scales::hue_pal()为其他调色板（如viridis::viridis_pal()）生成自定义颜色
• 使用purrr::map批量处理图表，避免重复编写代码
• 最终通过plot_layout(guides = "collect")将所有图例合并为一个

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