5

我在 R 中有一个函数可以采用一些默认值或自定义用户函数:

myfun = function(x = c('funA', 'funB', 'funC'), customFUN = NULL)
{
}

如果用户为 x 选择任何默认列出的值,customFUN 将被忽略。在 R 中处理这个问题的最佳方法是什么?

我可以使用

myfun = function(x = c('funA', 'funB', 'funC'), customFUN = NULL)
{

  if (! is.null(customFUN)) {
    choosenFUN = match.fun(customFUN)
  }
  else {
    choosenFUN.string = match.arg(x)

    if (choosenFUN == 'funA')
       choosenFUN = algorithmA
    else if (choosenFUN == 'funB')
       choosenFUN = algorithmB
    #...and so on for the other defaults
  }

}

但是添加额外的 arg customFUN 似乎很奇怪,也许有一个聪明的方法可以做到这一点。谢谢!

4

1 回答 1

0

我想我明白了,感谢评论:

myfun = function(x = c('funA', 'funB', 'funC'))
{
  if (is.function(x))
    func = x
  else                  
    func.string = switch(match.arg(x), funA = 'UseA', funB = 'UseB', funC = 'Usec')

  if (func.string == 'UseA') 
    func = algorithmA
  else if (func.string == 'UseB')
    func = algorithmB
  # and so on

  func
}
于 2012-11-10T01:43:42.670 回答