Curly curly操作：R自定义函数中如何获取传入变量名生成ID列

要实现将传入的变量名作为固定值填充到新列的所有行，需要先捕获传入的参数表达式，再转为字符串形式，而不是直接对{{var_name}}取names。

方案1：搭配rlang包的enquo+as_name（tidyverse风格推荐） #

library(dplyr)
library(rlang)

my_fun <- function(dat, var_name){
  dat %>%
    mutate(id_column = as_name(enquo(var_name)))
}

# 测试
my_fun(mtcars, cyl)

原理说明：enquo(var_name)会捕获传入函数的参数表达式，as_name()可以将这个表达式转为对应字符串，赋值到新列时会自动重复填充到所有行。curly curly语法({{}})本身是用来在dplyr动词内直接引用列取值，无法直接提取变量名字符串。

方案2：用base R的substitute+deparse（无需额外加载rlang） #

library(dplyr)

my_fun <- function(dat, var_name){
  var_str <- deparse(substitute(var_name))
  dat %>%
    mutate(id_column = var_str)
}

# 测试
my_fun(mtcars, cyl)

两种方案最终返回的结果一致，新增的id_column列所有行的值都是"cyl"。

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