自定义facet_grid函数中如何向labeller传递列名

问题场景 #

需要编写一个自定义ggplot2绘图函数，支持传入字符串格式的x/y轴列名、分面列名，以及对应分面变量的自定义标签。目前已解决facet_grid的公式生成问题，但labeller部分无法识别传入的字符串形式分面列名，导致自定义标签无法生效。

可正常运行的手动绘图代码 #

library(dplyr)
library(ggplot2)

# 示例数据
data(msleep)
filtered <- msleep %>% filter(conservation %in% c('en', 'vu', 'domesticated'))

# 自定义标签
fancy_labels_cons <- c('domesticated' = 'Domesticated', 'en' = 'Endangered', 'vu' = 'Vulnerable')
fancy_labels_vore <- c('carni' = 'Carnivore', 'herbi'='Herbivore', 'omni'='Omnivore', 'insecti'='Insectivore')

# 手动绘制带自定义标签的分面图
ggplot(filtered, aes(x = sleep_total, y = awake)) + 
  geom_point() +
  facet_grid(vore ~ conservation, labeller = labeller(conservation=fancy_labels_cons, vore=fancy_labels_vore))

失效的自定义函数代码 #

myfunc <- function(df, x, y, facetx, facety, label_x, label_y){
  # 将字符串转为分面公式
  facet_func <- as.formula(paste0(facetx, ' ~ ', facety))
  ggplot(df, aes(.data[[x]], .data[[y]])) + 
    geom_point() +
    facet_grid(facet_func, 
               # 此处失效：labeller识别的是参数名`facetx`而非其对应的字符串值（如conservation）
               labeller = labeller(facetx=label_x, facety=label_y))
}

# 调用函数
myfunc(filtered, x='sleep_total', y='awake', facetx='conservation', facety='vore', 
       label_x=fancy_labels_cons, label_y=fancy_labels_vore)

解决方案 #

问题核心是labeller需要接收以实际分面变量名为键的参数，而非函数内部的参数名（如facetx）。以下两种方法可实现动态配置：

方法1：构建命名列表+do.call调用labeller #

手动创建以分面变量名为键的参数列表，通过do.call将列表转为labeller的参数：

myfunc <- function(df, x, y, facetx, facety, label_facetx, label_facety){
  facet_func <- as.formula(paste0(facetx, ' ~ ', facety))
  # 构建labeller的参数列表，键为实际分面变量名
  labeller_args <- list()
  labeller_args[[facetx]] <- label_facetx
  labeller_args[[facety]] <- label_facety
  
  ggplot(df, aes(.data[[x]], .data[[y]])) + 
    geom_point() +
    facet_grid(facet_func, 
               labeller = do.call(labeller, labeller_args))
}

# 测试调用
myfunc(filtered, x='sleep_total', y='awake', 
       facetx='conservation', facety='vore',
       label_facetx=fancy_labels_cons, label_facety=fancy_labels_vore)

方法2：使用tidy eval动态指定参数（ggplot2 3.0+支持） #

利用tidy eval的!!sym()语法，将字符串变量名转为符号，再用:=动态设置labeller的参数名：

library(rlang) # 需要加载rlang包

myfunc <- function(df, x, y, facetx, facety, label_facetx, label_facety){
  facet_func <- as.formula(paste0(facetx, ' ~ ', facety))
  
  ggplot(df, aes(.data[[x]], .data[[y]])) + 
    geom_point() +
    facet_grid(facet_func, 
               labeller = labeller(!!sym(facetx) := label_facetx, !!sym(facety) := label_facety))
}

# 测试调用
myfunc(filtered, x='sleep_total', y='awake', 
       facetx='conservation', facety='vore',
       label_facetx=fancy_labels_cons, label_facety=fancy_labels_vore)

说明 #

• 方法1通过手动构建命名列表，确保labeller接收的参数名是实际的分面变量名（如conservation），而非函数内部的参数占位符。
• 方法2是ggplot2推荐的动态参数写法，利用tidy eval语法直接将字符串转为可识别的变量符号，更符合tidyverse风格。

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