R语言如何编写接收任意列名参数的可变参数自定义函数

问题复现 #

尝试编写支持不定数量参数的R函数，接收数据框和任意列名，通过within()拼接指定列生成新列时，直觉写法会触发列对象找不到的错误：

# 错误写法示例
example.fun <- function(input,...){
  res <- within(input,pasted <- paste(...))
  res
}

# 测试用例
df <- data.frame(x = c(1,2),y=c("a","b"))
example.fun(df,x,y)
# 报错：Error in paste(...) : object 'x' not found

临时使用attach()/detach()的方案存在明显缺陷：

• 写法冗余，函数异常中断时会导致数据框残留在搜索路径中
• 全局环境存在同名对象时会出现命名冲突，计算结果不可控

# 不推荐的临时方案
example.fun2 <- function(input,...){
  attach(input)
  res <- within(input,pasted <- paste(...))
  detach(input)
  res
}

错误原因 #

核心是R的非标准计算作用域规则问题：
...是外层函数的参数，直接在within()的表达式中调用paste(...)时，传入的列名（如x/y）会在外层函数的执行环境中求值，不会进入within()创建的数据框计算环境搜索列，因此R会去全局环境查找对应对象，找不到就触发报错。

标准实现方案 #

Base R 原生实现（无第三方依赖） #

只需要在within()的计算环境内将...转为列表，再通过do.call()传递给paste()，即可强制所有列名在数据框环境内解析，无任何副作用：

example.fun <- function(input, ..., sep = " ") {
  within(input, pasted <- do.call(paste, list(..., sep = sep)))
}

测试运行：

df <- data.frame(x = c(1,2), y = c("a","b"))
example.fun(df, x, y)
#   x y pasted
# 1 1 a    1 a
# 2 2 b    2 b

# 自定义分隔符测试
example.fun(df, x, y, sep = "_")
#   x y pasted
# 1 1 a    1_a
# 2 2 b    2_b

Tidyverse 生态实现（更灵活） #

如果使用dplyr处理数据，可以通过enquos()捕获...的未求值表达式，再用!!!拼接注入到paste()中，支持直接传入列计算表达式，适配更多复杂场景：

library(dplyr)

example.fun <- function(input, ..., sep = " ") {
  input %>% 
    mutate(pasted = paste(!!!enquos(...), sep = sep))
}

# 支持传入计算后的列
example.fun(df, x, toupper(y), sep = "-")
#   x y pasted
# 1 1 a    1-A
# 2 2 b    2-B

注意：所有方案都不需要修改搜索路径，不存在attach()带来的命名冲突、路径残留问题，是R语言官方推荐的非标准计算实现方式。

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