在函数调用中有条件地包含参数

flo*_*del 8 r function argument-passing

我想调用一个函数,但根据情况我可能会用额外的参数调用它.这是一个简单的例子:

FUN <- function(arg1 = "default1", arg2 = "default2", arg3 = "default3")
          print(list(arg1, arg2, arg3))


x1 <- "hi"
x2 <- TRUE
x3 <- 1:3

use.arg3 <- FALSE   # This will decide if `x3` is used or not.

if (use.arg3) {
   FUN(arg1 = x1, arg2 = x2, arg3 = x3)
} else {
   FUN(arg1 = x1, arg2 = x2)
}
Run Code Online (Sandbox Code Playgroud)

虽然代码清晰,但感觉有点多余.还想象一下,如果我有类似use.arg1use.arg2变量,我会有一个丑陋的可能性组合(8)...

我在下面发布了一个解决方案,但我发现它有点复杂,以至于我总是很难记住确切的语法.

如果您有更好的想法,感谢您的分享.

flo*_*del 12

一种解决方案是在使用do.call适当的参数构建列表后使用:

do.call(FUN, c(list(arg1 = x1, arg2 = x2),   # unconditional args
               list(arg3 = x3)[use.arg3]))   # conditional arg
Run Code Online (Sandbox Code Playgroud)

它可以很好地适用于多种条件:

do.call(FUN, c(list(arg1 = x1)[use.arg1]     # conditional arg
               list(arg2 = x2)[use.arg2]     # conditional arg
               list(arg3 = x3)[use.arg3]))   # conditional arg
Run Code Online (Sandbox Code Playgroud)

  • 为什么不组合它们,即`do.call(FUN,list(arg1 = x1,arg2 = x2,arg3 = x3)[c(use.arg1,use.arg2,use.arg3)])或者,我会做什么做一个矢量`use.args < - rep(TRUE,3)`并设置为FALSE我不想使用的任何参数,这给出了`do.call(FUN,list(arg1 = x1,arg2 = x2) ,ARG3 = X3)[use.args])`.其他可能性(使用`......等)取决于实际情况,但这是最通用的答案. (2认同)

jor*_*ran 7

继续关于弗洛尔德答案的评论的讨论,出于好奇,这是我的想法的延伸.不确定它是否真的是一个可行的选择:

FUN(arg1 = x1, arg2 = x2, arg3 = if(use.arg3) x3 else formals(FUN)$arg3)
Run Code Online (Sandbox Code Playgroud)