如何在ggplot2自定义函数中访问已有图层的信息？

自定义尺度函数可以通过访问ggplot对象的内部结构，获取图层中的数据信息（比如label1的类别数量），以下两种方法可以实现：

方法1：提前提取数据信息再传入尺度函数 #

先构建不含自定义尺度的基础绘图对象，从中提取目标变量的类别数量，再将该参数传入自定义尺度函数，按需调整调色板：

library(ggplot2)
library(tidyverse)

someData <- data.frame(x = 1:100 + runif(n = 100, min = -10, max = 10), 
                       y = 1:100 + runif(n = 100, min = -10, max = 10), 
                       label1 = sample(letters[1:10], size = 100, replace = T), 
                       label2 = sample(1:5, size = 100, replace = T)) %>%
    mutate(label2 = as.factor(label2))

# 构建基础ggplot对象（暂不添加自定义尺度）
base_plot <- ggplot(data = someData, aes(x = x, y = y, color = label1, shape = label2)) +
    geom_point()

# 提取label1的类别数量
label1_level_count <- length(unique(base_plot$data$label1))

# 扩展自定义尺度函数，接收类别数量参数
scale_color_shapely <- function(n_levels, ...) {
    # 根据类别数量调整调色板参数，这里示例调整色调范围
    custom_pal <- scales::hue_pal(h = c(0, 360) - 15, c = 80, l = 60)(n_levels)
    discrete_scale("color", scale_name = "blaaa", palette = function(n) custom_pal, ...)
}

# 将自定义尺度添加到基础图中
final_plot <- base_plot +
    scale_color_shapely(n_levels = label1_level_count, name = "myDummyColorScale")

print(final_plot)

方法2：让尺度函数自动解析ggplot对象 #

如果希望尺度函数能自动从ggplot对象中提取数据信息，可以让函数接收ggplot对象作为参数，内部通过ggplot_build()解析绘图数据：

scale_color_shapely <- function(gg_obj, ...) {
    # 解析ggplot对象，获取绘图用的数据集
    built_plot <- ggplot_build(gg_obj)
    # 提取color映射对应的变量（这里是label1）
    color_var <- gg_obj$mapping$color
    # 提取该变量的所有取值并统计类别数量
    color_data <- pull(built_plot$data[[1]], {{color_var}})
    n_levels <- length(unique(color_data))
    
    # 根据类别数量生成自定义调色板
    custom_pal <- scales::hue_pal(h = c(0, 300), c = 90, l = 55)(n_levels)
    discrete_scale("color", scale_name = "blaaa", palette = function(n) custom_pal, ...)
}

# 使用方式：先构建基础图，再传入尺度函数
base_plot <- ggplot(data = someData, aes(x = x, y = y, color = label1, shape = label2)) +
    geom_point()

final_plot <- base_plot +
    scale_color_shapely(base_plot, name = "myDummyColorScale")

print(final_plot)

核心要点 #

• ggplot_build()会将ggplot对象转换为包含实际绘图数据、图层配置的结构，built_plot$data[[1]]对应第一个图层的数据集。
• {{color_var}}利用rlang的非标准求值语法，直接提取映射中指定的color变量对应的数据。
• 你可以基于提取到的类别数量，灵活调整调色板的色调、饱和度、亮度等参数，实现更适配的自定义颜色尺度。

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