ggplot2自定义geom_实现双样式置信区间绘制的方案求助

报错根因 #

在stat_my_confint中默认传入的geom = "my_confint"为字符串类型，ggplot2接收字符串格式的geom参数时，会自动在全局环境搜索命名规则为Geom[字符串首字母大写后缀]的ggproto对象，当前仅实现了geom_my_confint构造函数，未定义对应的GeomMyConfint类，因此匹配失败抛出错误。

可行解决方案 #

方案1：最小改动适配stat调用（推荐） #

直接修改stat_my_confint的逻辑，和geom_my_confint保持一致，根据style参数动态选择内置的Geom对象，无需传入字符串类型的geom名称，即可兼容直接调用stat的场景：

stat_my_confint <- function(mapping = NULL, data = NULL, 
                            position = "identity", na.rm = FALSE,
                            show.legend = NA, inherit.aes = TRUE,
                            style = c("polygon", "line"), ...) {
  style <- match.arg(style)
  # 动态选择对应Geom对象，绕过字符串匹配逻辑
  target_geom <- if (style == "line") ggplot2::GeomPath else ggplot2::GeomPolygon
  ggplot2::layer(
    geom = target_geom, 
    stat = StatMyConfint,
    data = data,
    mapping = mapping,
    position = position,
    show.legend = show.legend,
    inherit.aes = inherit.aes,
    params = list(
      na.rm = na.rm,
      style = style,
      ...
    )
  )
}

修改完成后，原本报错的stat_my_confint()调用即可正常运行，也支持传入style = "line"切换绘制样式，无需调整其他代码。

方案2：完善自定义Geom类实现 #

如果需要保留统一的Geom入口，可优化原思路ii的实现，通过重写use_defaults方法解决动态default_aes的问题：

1. 首先定义自定义Geom类，根据style参数动态加载对应绘制类型的默认美学参数：

GeomMyConfint <- ggplot2::ggproto("GeomMyConfint", ggplot2::Geom,
  required_aes = c("x", "y"),
  use_defaults = function(self, data, params = list()) {
    style <- params$style %||% "polygon"
    # 动态匹配对应geom的默认aes配置
    target_geom <- if (style == "line") ggplot2::GeomPath else ggplot2::GeomPolygon
    self$default_aes <- target_geom$default_aes
    NextMethod()
  },
  draw_panel = function(data, panel_params, coord, style) {
    target_geom <- if (style == "line") ggplot2::GeomPath else ggplot2::GeomPolygon
    target_geom$draw_panel(data, panel_params, coord)
  }
)

2. 修改geom_my_confint的layer构造逻辑，使用自定义的GeomMyConfint作为geom参数：

geom_my_confint <- function(mapping = NULL, data = NULL, stat = "my_confint",
                            position = "identity", na.rm = FALSE,
                            show.legend = NA, inherit.aes = TRUE,
                            style = c("polygon", "line"), ...) {
  style <- match.arg(style)
  ggplot2::layer(
    geom = GeomMyConfint,
    mapping = mapping,
    data = data,
    stat = stat,
    position = position,
    show.legend = show.legend,
    inherit.aes = inherit.aes,
    params = list(
      na.rm = na.rm,
      style = style,
      ...
    )
  )
}

该方案下stat_my_confint无需修改，默认传入的geom = "my_confint"可正常匹配到GeomMyConfint类，同时解决了动态默认美学设置的问题，实现更符合ggplot2的扩展规范。

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