ggplot2自定义绘图函数灵活配置aes内外美学参数的实现方案

可行解决方案 #

你可以通过条件判断动态生成geom_point的美学参数：只有当用户没有传入映射变量（.shapefac/.colfac为NULL）时，才把固定样式参数传到aes()外部，否则只使用aes()内部的映射逻辑，就不会出现覆盖问题。

原代码的核心问题是不管是否传入映射变量，都把固定样式参数写到了geom_point的顶层，而ggplot2的规则就是顶层固定参数优先级高于aes()内部映射，必然会覆盖映射结果。下面的修改无需调整ggplot2本身的默认规则，也不需要使用...传递参数，保持了函数接口的易用性：

library(mlbench)
library(ggplot2)
data("Ionosphere")

cplot <- function(X, v = 1:ncol(X), .shapefac = NULL, .colfac = NULL, shape = 21, colour = "black",
                  center = TRUE, scale = FALSE, x = 1, y = 2, plot = TRUE) {
  
  # 数据预处理逻辑保持不变
  d.pca <- prcomp(X, center = center, scale. = scale)
  Y <- data.frame(X, d.pca$x)
  var_exp <- round(100 * (d.pca$sdev^2 / sum(d.pca$sdev^2)), 2)
  
  # 基础绘图对象
  p <- ggplot(Y, aes(x = .data[[paste0("PC",x)]], y = .data[[paste0("PC",y)]])) 
  
  # 动态构造点图层参数
  point_aes <- list()
  point_fixed <- list()
  
  # 处理shape逻辑：有映射就放进aes，没有就用固定值
  if(!is.null(.shapefac)) {
    point_aes[["shape"]] <- .data[[.shapefac]]
  } else {
    point_fixed[["shape"]] <- shape
  }
  
  # 处理colour逻辑：有映射就放进aes，没有就用固定值
  if(!is.null(.colfac)) {
    point_aes[["colour"]] <- .data[[.colfac]]
  } else {
    point_fixed[["colour"]] <- colour
  }
  
  # 拼接参数生成点图层
  p <- p +
    do.call(geom_point, c(list(mapping = do.call(aes, point_aes)), point_fixed)) +
    labs(x = paste0("PC ", x, " (", var_exp[x], "%)"),
         y = paste0("PC ", y, " (", var_exp[y], "%)")) +
    theme_bw()
  
  if (plot) print(p)
  invisible(p)
}

功能验证 #

所有场景均可正常运行：

1. 仅指定固定样式：cplot(Ionosphere, v = 3:34, colour = "tomato", shape = 4)
2. 仅指定shape列映射：cplot(Ionosphere, v = 3:34, .shapefac = "Class")
3. 仅指定colour列映射：cplot(Ionosphere, v = 3:34, .colfac = "Class")
4. 同时指定两个映射：cplot(Ionosphere, v = 3:34, .shapefac = "Class", .colfac = "Class")

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