ggplot2自定义geom_linepoint未指定group美学时无法正常工作

问题根源 #

ggplot的自动分组逻辑会将所有离散型美学变量（包括你自定义的disp.pt逻辑向量）作为分组依据。当未显式指定group时，原本属于同一线条的数据会被拆分为disp.pt=TRUE和disp.pt=FALSE两组，部分组可能因数据量不足（比如仅单个点或空数据），导致unit()函数接收到空参数触发报错。而原生geom_line不存在这个问题，因为它的美学体系里没有disp.pt，不会触发额外的分组拆分。

修复方案 #

方案1：在setup_data中强制统一分组 #

通过自定义geom的setup_data方法，当用户未指定group时，手动为所有数据分配同一个分组ID，覆盖ggplot的自动分组逻辑：

# 自定义geom_linepoint的完整实现
geom_linepoint <- function(mapping = NULL, data = NULL, stat = "identity",
                           position = "identity", na.rm = FALSE, show.legend = NA,
                           inherit.aes = TRUE, ...) {
  layer(
    data = data,
    mapping = mapping,
    stat = stat,
    geom = GeomLinepoint,
    position = position,
    show.legend = show.legend,
    inherit.aes = inherit.aes,
    params = list(na.rm = na.rm, ...)
  )
}

GeomLinepoint <- ggproto("GeomLinepoint", GeomLine,
  required_aes = c("x", "y", "disp.pt"),
  default_aes = aes(colour = "black", size = 1, linetype = 1, alpha = NA,
                    shape = 19, fill = NA),
  
  # 重写setup_data处理分组
  setup_data = function(data, params) {
    # 未指定group时，统一设置为同一个分组
    if (is.null(data$group)) {
      data$group <- 1L
    }
    # 调用父类GeomLine的原始setup_data逻辑
    GeomLine$setup_data(data, params)
  },
  
  draw_panel = function(data, panel_params, coord, na.rm = FALSE) {
    # 先绘制线条
    line_grob <- GeomLine$draw_panel(data, panel_params, coord, na.rm)
    # 筛选需要显示的点并绘制
    point_data <- data[data$disp.pt, , drop = FALSE]
    if (nrow(point_data) > 0) {
      point_grob <- GeomPoint$draw_panel(point_data, panel_params, coord, na.rm)
      grobTree(line_grob, point_grob)
    } else {
      line_grob
    }
  }
)

方案2：在draw_panel中重分组数据 #

如果需要保留其他美学变量（如colour、linetype）的分组逻辑，仅排除disp.pt的影响，可以在draw_panel中重新分组：

draw_panel = function(data, panel_params, coord, na.rm = FALSE) {
  # 按除disp.pt外的离散美学变量重新分组
  data <- data %>%
    dplyr::group_by(dplyr::across(-c(disp.pt, x, y))) %>%
    dplyr::mutate(group = dplyr::cur_group_id()) %>%
    dplyr::ungroup()
  
  # 后续绘制逻辑不变
  line_grob <- GeomLine$draw_panel(data, panel_params, coord, na.rm)
  point_data <- data[data$disp.pt, , drop = FALSE]
  if (nrow(point_data) > 0) {
    point_grob <- GeomPoint$draw_panel(point_data, panel_params, coord, na.rm)
    grobTree(line_grob, point_grob)
  } else {
    line_grob
  }
}

验证测试 #

用以下代码验证修复效果（无需指定group）：

set.seed(123)
df <- data.frame(
  x = 1:10,
  y = rnorm(10),
  disp.pt = sample(c(TRUE, FALSE), 10, replace = TRUE)
)

# 未指定group，应正常生成图形
ggplot(df, aes(x, y, disp.pt = disp.pt)) + geom_linepoint()

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