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

时间:2012-12-21 14:59:55

标签: 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)
}

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

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

如果你有更好的主意,谢谢分享。

2 个答案:

答案 0 :(得分:10)

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

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

它可以很好地概括为多种条件:

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

答案 1 :(得分:6)

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

FUN(arg1 = x1, arg2 = x2, arg3 = if(use.arg3) x3 else formals(FUN)$arg3)