为多幅含唯一因子水平组合的ggplot图创建通用图例

问题背景 #

我有多幅柱状图，每幅对应某一通用因子的唯一水平组合，需要为所有图表设置一个共享图例以展示该因子的全部水平。现有代码能实现需求，但在处理大量图表或大量因子水平的场景下过于繁琐，寻求更优方案。

现有基础代码 #

library(ggplot2)
library(grid)
library(gridExtra)

set.seed(1)
df1 <- data.frame(c=c('a', 'b', 'c', 'd', 'e'), x=1:5,  y=runif(5))
df2 <- data.frame(c=c('a', 'c', 'e', 'f', 'g'), x=1:5,  y=runif(5))

dd <- union(df1$c,df2$c) #### 合并所有因子水平
dd.col <- rainbow(length(dd)) ### 自定义颜色向量
names(dd.col)  <- dd #### 为颜色向量命名，对应因子水平

g1 <- ggplot(df1, aes(x=x, y=y, fill=c)) + 
  geom_bar(stat="identity") +
  scale_fill_manual("Legend", values = dd.col)
g2 <- ggplot(df2, aes(x=x, y=y, fill=c)) + 
  geom_bar(stat="identity") +
  scale_fill_manual("Legend", values = dd.col)
grid.arrange(g1, g2, ncol=2)

现有繁琐方案（不适用于大量图表/因子） #

df3 <- data.frame(c=c('a', 'b', 'c', 'd', 'e', 'f', 'g'), x=1:7,  y=runif(7))
g3 <- ggplot(df3, aes(x=x, y=y, fill=c)) + 
  geom_bar(stat="identity") +
  scale_fill_manual("Legend", values = dd.col) +
  theme(legend.position = "bottom")

library(cowplot)

leg <- get_legend(g3)
plot_grid(
  plot_grid(g1 + theme(legend.position = "none"),
            g2 + theme(legend.position = "none"), ncol=2),
          leg,
          align = "hv", nrow=2, rel_heights = c(1,0.1))

高效解决方案 #

方法1：用patchwork包简化布局与图例提取 #

patchwork是ggplot2生态中专门用于组合图表的工具，语法简洁，处理多图表场景效率更高：

library(ggplot2)
library(patchwork)

# 批量移除子图的图例
g1_noleg <- g1 + theme(legend.position = "none")
g2_noleg <- g2 + theme(legend.position = "none")

# 提取完整共享图例（无需额外构造数据集，直接用已定义好scale的子图即可）
shared_legend <- get_legend(g1 + theme(legend.position = "bottom"))

# 组合图表与图例，按比例分配高度
(g1_noleg + g2_noleg) / shared_legend + plot_layout(heights = c(9, 1))

说明：因为scale_fill_manual已经指定了所有因子水平的颜色，即使子图只包含部分水平，提取任意子图的图例就能得到完整的共享图例。

方法2：数据合并后用facet分面（最优结构化方案） #

如果所有数据集可以合并并添加分组标识，直接用ggplot的分面功能，自动生成共享图例，这是代码量最少的方案：

library(ggplot2)

# 合并所有数据集并添加图表分组标识
df_combined <- rbind(
  cbind(df1, group = "图1"),
  cbind(df2, group = "图2")
)

# 绘制分面柱状图，自动共享图例
ggplot(df_combined, aes(x=x, y=y, fill=c)) +
  geom_bar(stat="identity") +
  scale_fill_manual("Legend", values = dd.col) +
  facet_wrap(~group, ncol=2) +
  theme(legend.position = "bottom")

优点：无需手动组合多个独立图表，只需维护一个合并后的数据集，适合大量图表的批量生成场景。

方法3：自定义函数批量处理独立图表 #

如果必须保留独立的图表对象，可通过自定义函数批量移除图例、提取共享图例并组合：

library(ggplot2)
library(gridExtra)
library(gtable)

# 批量移除图例的工具函数
remove_legend <- function(p) {
  p + theme(legend.position = "none")
}

# 假设你有一个包含所有图表的列表（可批量生成）
plots <- list(g1, g2)
plots_noleg <- lapply(plots, remove_legend)

# 提取共享图例
shared_legend <- gtable_filter(ggplotGrob(g1), "guide-box")

# 组合所有无图例图表与共享图例
grid.arrange(
  arrangeGrob(grobs = plots_noleg, ncol=2),
  shared_legend,
  nrow=2, heights = c(9, 1)
)

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