基于 R 中不同的参数名称创建函数

Question

我想创建一个基于不同参数和主体生成函数的函数。

例如，取n=2 和a=paste0("a",1:2n)，即a=c("a1","a2","a3","a4")。

那我想创建一个函数如下-

fa = function(a1,a2,a3,a4) { 
args_fa = unlist(as.list(environment())) # catch all the arguments as a list
v1 = args_fa[1:n] # assign values to baseline function arguments
v2 = args_fa[(n+1):2*n]
f(v1,v2)
} 

n=3 也一样，我希望函数看起来像这样 -

fa = function(a1,a2,a3,a4,a5,a6) { 
args_fa = unlist(as.list(environment())) # catch all the arguments as a list
v1 = args_fa[1:n] # assign values to baseline function arguments
v2 = args_fa[(n+1):2*n]
f(v1,v2)
} 

函数内部的f 是一个基线函数（固定），不会发生变化。我的问题是如何根据更改n 创建fa？

Answer 1

使用省略号...，并在参数列表中包含n：

fa <- function(n, ...) {
  stopifnot(length(n) == 1L, n > 0)
  args_fa <- unlist(list(...))
  v1 <- args_fa[1:n]
  v2 <- args_fa[(n+1):(2*n)]
  f(v1, v2)
}

或者，将n 放在最后。这要求 n 是一个命名参数，它不能作为位置参数传递。

fa <- function(..., n) {
  stopifnot(length(n) == 1L, n > 0)
  args_fa <- unlist(list(...))
  v1 <- args_fa[1:n]
  v2 <- args_fa[(n+1):(2*n)]
  f(v1, v2)
}

或者如果你总是有一个向量或参数列表，那么也许

fa <- function(args_fa, n) {
  stopifnot(length(n) == 1L, n > 0)
  v1 <- args_fa[1:n]
  v2 <- args_fa[(n+1):(2*n)]
  f(v1, v2)
}