如何为geom_grob的自定义图形添加图例？

geom_grob本身不会自动为自定义grob生成图例，因为它默认的图例绘制函数（draw_key）无法识别并渲染你的grob对象。要实现自定义grob的图例显示，需要自定义图例绘制函数，并绑定到geom_grob的参数中，以下是具体实现步骤：

1. 重构数据与grob生成逻辑 #

先将数据中的grob对象替换为分类变量（如square/triangle），同时封装生成grob的函数，方便后续调用和颜色映射：

library(grid)
library(ggplot2)
library(dplyr)
library(ggpp)

# 封装生成自定义grob的函数
make_shape_grob <- function(type, color = "blue") {
  devratio <- dev.size()[2]/dev.size()[1]
  gp <- gpar(lwd = 2, fill = color, col = color)
  
  if (type == "square") {
    square_ang <- seq(pi/4, 9*pi/4, length.out = 5)
    grid.polygon(
      x = unit(0.5 + 0.2 * devratio * sin(square_ang), 'npc'),
      y = unit(0.5 + 0.2 * cos(square_ang), 'npc'),
      gp = gp
    )
  } else if (type == "triangle") {
    triangle_ang <- seq(0, 2*pi, length.out = 4)
    grid.polygon(
      x = unit(0.5 + 0.18 * devratio * sin(triangle_ang), 'npc'),
      y = unit(0.5 + 0.18 * cos(triangle_ang), 'npc'),
      gp = gp
    )
  }
}

# 构造规范数据集
df <- tibble(
  x = 1:6, 
  y = 1:6, 
  shape_type = rep(c("square", "triangle"), each = 3),
  col = rep(c('col1', 'col2'), each = 3)
)

2. 自定义图例绘制函数 #

创建draw_key函数，负责在图例框内渲染对应的自定义grob：

draw_key_custom_grob <- function(data, params, size) {
  # 根据图例数据生成对应grob
  grob <- make_shape_grob(data$shape_type, data$colour)
  # 将grob适配到图例的标准尺寸中
  grid::grobTree(grob, vp = grid::viewport(width = size, height = size))
}

3. 绘图并绑定自定义图例函数 #

在geom_grob中指定自定义的grob生成逻辑和draw_key函数，同时开启图例显示：

ggplot(df) +
  geom_grob(
    aes(x = x, y = y, label = shape_type, color = col),
    # 让grob颜色跟随color美学映射
    grob = function(x) make_shape_grob(x, color = ggplot2::after_scale(colour)),
    show.legend = TRUE,
    draw_key = draw_key_custom_grob
  ) +
  labs(color = "类别", shape = "图形")

关键说明 #

• 直接传递grob列表给label时，ggplot无法将其识别为图例分类依据，必须使用可映射的分类变量（如字符串）。
• 默认draw_key函数不支持自定义grob的渲染，因此需要手动实现图例框内的绘制逻辑。
• after_scale(colour)用于让grob的颜色跟随color美学的映射值，替代固定颜色设置。

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