如何让facet_wrap绘图的图例色条高度匹配面板区域？

使用facet_wrap()绘制多面板图时，常规调整图例色条高度的方法会导致色条延伸到分面标题和轴文本区域，我们需要让色条仅覆盖从分面标题底部到轴文本顶部的面板区域。

解决方案 #

核心是精准提取分面图中所有面板的总高度，而非简单减去非面板区域的高度。修改自定义函数，定位面板行并计算其总高度，以此设置色条高度：

library(tidyverse)
library(reshape2)

# 定义自定义函数，匹配面板高度
make_fullsize <- function() structure("", class = "fullsizebar")

ggplot_add.fullsizebar <- function(obj, g, name = "fullsizebar") {
  grob <- ggplotGrob(g)
  # 定位所有面板所在的行
  panel_rows <- unique(grob$layout$t[grep("panel", grob$layout$name)])
  # 提取这些行的高度总和
  total_panel_height <- sum(grob$heights[panel_rows])
  
  g + 
    guides(fill = guide_colorbar(barheight = total_panel_height,
                                 title.position = "right")) +
    theme(legend.title = element_text(angle = -90, hjust = 0.5))
}

# 绘制分面相关矩阵图
iris %>%
  group_by(Species) %>%
  summarise(correlation = list(cor(across(where(is.numeric))))) %>%
  ungroup() %>%
  mutate(correlation_df = map(correlation, ~ melt(.x, varnames = c("Var1", "Var2")))) %>%
  select(Species, correlation_df) %>%
  unnest(correlation_df) %>%
  ggplot(aes(x = Var1, y = Var2, fill = value)) +
  geom_tile() +
  scale_fill_gradient2(limit = c(-1, 1)) +
  labs(x = NULL, y = NULL, fill = NULL) +
  theme_bw() +
  coord_cartesian(expand = FALSE) +
  make_fullsize() +
  facet_wrap(~Species)

原理说明 #

1. 通过ggplotGrob()提取绘图的底层grob结构，找到所有面板对应的行位置
2. 计算这些面板行的高度总和，直接作为色条的barheight参数
3. 这种方式会让色条高度精确匹配分面面板的总高度，不会延伸到标题或轴文本区域，且调整绘图大小时能自动适配

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