如何通过编程获取ggplot对象的图例行数？

要获取ggplot对象的图例行数，需要借助ggplot_gtable()提取图表的布局信息（ggplot_build()仅处理数据层，不包含图例布局细节），具体步骤如下：

• 第一步：将ggplot对象转换为gtable对象，该对象包含图表所有元素的布局信息
• 第二步：定位到图例对应的gtable组件
• 第三步：从图例组件的布局中读取行数

完整代码示例 #

library(dplyr)
library(ggplot2)

# 生成示例绘图对象
p1 <- mpg %>% 
  count(drv, class) %>% 
  ggplot(aes(drv, n, fill = class)) +
  geom_col() +
  scale_fill_viridis_d(NULL, option = "H") +
  ggtitle("class") +
  theme_minimal() +
  theme(legend.position = "bottom")

p2 <- mpg %>% 
  count(drv, manufacturer) %>% 
  ggplot(aes(drv, n, fill = manufacturer)) +
  scale_fill_viridis_d(NULL, option = "H") +
  ggtitle("manufacturer") +
  theme_minimal() +
  theme(legend.position = "bottom")

# 定义获取图例行数的函数
get_legend_rows <- function(p) {
  # 转换为gtable
  gt <- ggplot_gtable(ggplot_build(p))
  # 定位图例组件
  legend_idx <- which(sapply(gt$grobs, function(x) x$name) == "guide-box")
  if (length(legend_idx) == 0) {
    return(0) # 无图例时返回0
  }
  legend_grob <- gt$grobs[[legend_idx]]
  # 返回图例行数
  nrow(legend_grob$layout)
}

# 测试函数
get_legend_rows(p1) # 输出：2
get_legend_rows(p2) # 输出：3

说明 #

• ggplot_gtable()会基于ggplot_build()的结果生成完整的图表布局结构，其中包含了图例的排列行数/列数信息
• 通过sapply(gt$grobs, function(x) x$name)可以筛选出名为guide-box的图例组件
• 图例组件的layout属性记录了其网格布局，nrow()即可获取行数

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