如何从R语言ggplot2的aes()映射中提取变量名

假设我们有如下ggplot2映射：

mapping = ggplot2::aes(x=ifelse(a, 1, 2), y=a*b)
mapping

运行后输出：

Aesthetic mapping:

• x -> ifelse(a, 1, 2)
• y -> a * b

该映射形式不固定，我需要提取其中引用的变量名，用于获取数据并构建数据框（因数据量限制，仅保留可视化必需的列）。目标是实现函数f()，满足：

f(mapping) == c('a', 'b')

目前已能通过以下代码提取表达式文本：

expressions = sapply(mapping, quo_name)
expressions

输出：

x                 y 
"ifelse(a, 1, 2)"           "a * b" 

但不清楚如何从这些字符串中提取变量/列名。

解决方案 #

直接操作mapping中的quosure对象（而非解析字符串）是更可靠的方式，具体实现如下：

library(rlang)

f <- function(mapping) {
  # 遍历每个映射项，提取其中的变量符号
  extracted_vars <- lapply(mapping, function(quo) {
    # 获取quosure对应的表达式
    expr <- rlang::get_expr(quo)
    # 提取表达式中所有名称
    all_names <- all.names(expr)
    # 排除函数名，仅保留变量名
    setdiff(all_names, all.functions(expr))
  })
  
  # 合并结果、去重后返回
  unique(unlist(extracted_vars))
}

测试验证 #

运行示例代码：

mapping = ggplot2::aes(x=ifelse(a, 1, 2), y=a*b)
f(mapping)
# 输出：[1] "a" "b"

对于更复杂的映射表达式，比如：

mapping2 = ggplot2::aes(x=log(c) + d*sin(e), y=f/g)
f(mapping2)
# 输出：[1] "c" "d" "e" "f" "g"

该函数也能准确提取所有变量名。

这种方法通过直接解析R的表达式结构，避免了字符串解析可能带来的格式兼容问题，鲁棒性更强。

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