为多幅不同筛选条件的ggplot图表添加统一图例

嘿，我来帮你搞定这个统一图例的问题！你已经有了三个基于不同子集的直方图，现在需要一个包含所有cut因子的统一图例，下面是完整的可运行方案：

完整代码实现 #

# 加载需要的包
library(ggplot2); library(gridExtra); library(grid)
# 如果用get_legend的方法，需要加载cowplot（可选，也可以用自定义函数）
# library(cowplot)

# 准备数据子集和颜色映射
z <- subset(diamonds, subset = cut == "Fair" | cut == "Good")
y <- subset(diamonds, subset = cut == "Good")
x <- subset(diamonds, subset = cut == "Premium" | cut == "Fair")
colours = c("Fair" = "#666362", "Good" = "#D40511", "Very Good" = "#FFCC00", "Premium" = "#000000", "Ideal" = "#BBBBB3")

# 创建三个直方图，**隐藏各自的不完整图例**
my_hist_z <- ggplot(z, aes(clarity, fill=cut)) + 
  geom_bar() + 
  scale_fill_manual(values = colours) +
  theme(legend.position = "none")

my_hist_y <- ggplot(y, aes(clarity, fill=cut)) + 
  geom_bar() + 
  scale_fill_manual(values = colours) +
  theme(legend.position = "none")

my_hist_x <- ggplot(x, aes(clarity, fill=cut)) + 
  geom_bar() + 
  scale_fill_manual(values = colours) +
  theme(legend.position = "none")

# 方法1：用自定义函数提取完整图例（无需额外包）
g_legend <- function(a.gplot){
  tmp <- ggplot_gtable(ggplot_build(a.gplot))
  leg <- which(sapply(tmp$grobs, function(x) x$name) == "guide-box")
  legend <- tmp$grobs[[leg]]
  return(legend)
}

# 创建包含所有cut因子的图例源图
full_legend_plot <- ggplot(data.frame(cut = names(colours)), aes(fill=cut)) +
  geom_bar() +
  scale_fill_manual(values = colours)
# 提取图例对象
legend <- g_legend(full_legend_plot)

# 方法2：用cowplot的get_legend（更简洁）
# legend_plot <- ggplot(data.frame(cut = names(colours)), aes(x=1, y=1, fill=cut)) +
#   geom_rect() +
#   scale_fill_manual(values = colours) +
#   theme_void() +
#   theme(legend.position = "bottom")
# legend <- get_legend(legend_plot)

# 组合图表和统一图例
grid.arrange(
  # 先把三个直方图排成一行
  grid.arrange(my_hist_z, my_hist_y, my_hist_x, ncol=3),
  # 再放上统一图例
  legend,
  nrow=2,
  heights=c(4,1) # 调整图和图例的高度比例，让布局更协调
)

关键步骤解释 #

• 隐藏子图图例：每个直方图基于的子集只包含部分cut因子，自带的图例是不完整的，所以我们用theme(legend.position = "none")把它们都隐藏，避免出现多个零散的图例。
• 生成统一图例：我们创建了一个包含所有5个cut因子的数据集（直接复用颜色向量的名称），基于这个数据集生成一个图例源图，再提取出单独的图例对象。
• 布局组合：用grid.arrange先把三个直方图横向排列，再在下方放上统一的图例，通过heights参数调整图和图例的占比，让整体布局更美观。

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