如何在ggpairs自定义函数中访问面板的行、列特定变量？

核心问题 #

GGally::wrap()函数默认不会将ggpairs内部的面板行/列索引（i, j）传递给自定义绘图函数，导致无法生成依赖于位置的标注。

解决方案1：直接使用自定义函数（无需wrap） #

ggpairs在调用面板函数时，会自动将当前面板的i（行索引）和j（列索引）作为参数传递，只要自定义函数的签名包含这两个参数，就能直接使用。同时可以根据数据范围调整标注位置，确保文本可见：

library(data.table)
library(ggplot2)
library(GGally)

# 生成示例数据
dt <- as.data.table(diamonds)[sample(.N, 50), .(carat, depth, x, y, z)]

# 定义带行/列标注的自定义绘图函数
custom_plot <- function(data, mapping, i, j, ...) {
  # 获取当前x、y轴的数据范围，用于定位文本
  x_var <- rlang::get_expr(mapping$x)
  y_var <- rlang::get_expr(mapping$y)
  x_range <- range(data[[x_var]], na.rm = TRUE)
  y_range <- range(data[[y_var]], na.rm = TRUE)
  
  ggplot(data = data, mapping = mapping) + 
    geom_point(...) +
    # 左上角标注行索引
    annotate("text", x = x_range[1], y = y_range[2], 
             label = paste0("Row: ", i), hjust = 0, vjust = 1, size = 4) + 
    # 右下角标注列索引
    annotate("text", x = x_range[2], y = y_range[1], 
             label = paste0("Column: ", j), hjust = 1, vjust = 0, size = 4) + 
    theme_minimal()
}

# 生成ggpairs图，直接指定自定义函数（不用wrap）
ggpairs(dt,
        upper = list(continuous = wrap(pcplot)),
        diag = list(continuous = "blank"),
        lower = list(continuous = custom_plot)
)

解决方案2：自定义wrap函数（支持传递额外参数） #

如果需要给自定义函数传递额外参数，同时保留i/j的传递，可以自己实现一个包装函数，手动传递索引参数：

library(data.table)
library(ggplot2)
library(GGally)

dt <- as.data.table(diamonds)[sample(.N, 50), .(carat, depth, x, y, z)]

custom_plot <- function(data, mapping, i, j, point_size = 1.5, text_size = 4, ...) {
  x_var <- rlang::get_expr(mapping$x)
  y_var <- rlang::get_expr(mapping$y)
  x_range <- range(data[[x_var]], na.rm = TRUE)
  y_range <- range(data[[y_var]], na.rm = TRUE)
  
  ggplot(data = data, mapping = mapping) + 
    geom_point(size = point_size, ...) +
    annotate("text", x = x_range[1], y = y_range[2], 
             label = paste0("Row: ", i), hjust = 0, vjust = 1, size = text_size) + 
    annotate("text", x = x_range[2], y = y_range[1], 
             label = paste0("Column: ", j), hjust = 1, vjust = 0, size = text_size) + 
    theme_minimal()
}

# 自定义wrap函数，传递i/j和额外参数
custom_wrap <- function(func, ...) {
  extra_args <- list(...)
  function(data, mapping, i, j) {
    do.call(func, c(list(data = data, mapping = mapping, i = i, j = j), extra_args))
  }
}

# 调用ggpairs，使用自定义wrap传递参数
ggpairs(dt,
        upper = list(continuous = wrap(pcplot)),
        diag = list(continuous = "blank"),
        lower = list(continuous = custom_wrap(custom_plot, point_size = 2, text_size = 3.5))
)

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