R语言中如何基于可变参数名动态生成指定结构的函数

我们可以通过动态构造函数形参列表+函数表达式的方式实现需求，优先推荐不需要额外依赖的基础R实现方案：

generate_fa <- function(n) {
  # 1. 生成参数名向量 a1 ~ a2n
  arg_names <- paste0("a", seq_len(2*n))
  # 2. 构造空的形参列表
  args <- setNames(rep(list(quote(expr = )), length(arg_names)), arg_names)
  # 3. 构造函数体表达式，捕获当前n值避免后续修改影响已生成的函数
  body <- bquote({
    args_fa = unlist(as.list(environment()))
    v1 = args_fa[1:.(n)]
    v2 = args_fa[.(n+1):.(2*n)]
    f(v1, v2)
  })
  # 4. 组装生成目标函数并返回
  fa <- as.function(c(args, body))
  return(fa)
}

效果验证示例 #

先定义测试用的固定基线函数f：

# 示例基线函数：返回两组参数的求和结果
f <- function(v1, v2) {
  sum(v1) + sum(v2)
}

n=2场景测试 #

fa_n2 <- generate_fa(n = 2)
# 查看生成的函数结构，可确认参数为a1/a2/a3/a4，逻辑符合要求
print(fa_n2)
# 调用测试：传入a1=1,a2=2,a3=3,a4=4，预期返回结果1+2+3+4=10
fa_n2(1,2,3,4)

n=3场景测试 #

fa_n3 <- generate_fa(n = 3)
print(fa_n3)
# 调用测试：传入a1~a6为1-6，预期返回结果1+2+3+4+5+6=21
fa_n3(1,2,3,4,5,6)

可选rlang包实现方案 #

如果你习惯使用rlang的元编程接口，也可以用更简洁的写法实现：

library(rlang)
generate_fa_rlang <- function(n) {
  arg_names <- paste0("a", seq_len(2*n))
  args <- set_names(rep(list(missing_arg()), length(arg_names)), arg_names)
  body <- expr({
    args_fa = unlist(as.list(environment()))
    v1 = args_fa[1:!!n]
    v2 = args_fa[!!(n+1):!!(2*n)]
    f(v1, v2)
  })
  new_function(args, body, env = current_env())
}

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