r - R代码：使用带有多个参数的UDF用于应用函数

Question

我的UDF：

testfn = function(x1, x2, x3){
if(x1 > 0){y = x1 + x2 + x3}
if(x1 < 0){y = x1 - x2 - x3}
return(y)
}

我的样本测试集：

test = cbind(rep(1,3),c(2,4,6),c(1,2,3))

申请的运行：

apply(test, 1, testfn, x1 = test[1], x2 = test[2], x3 = test[3])

这是我得到的错误：

FUN(newX[, i], ...) 中的错误：未使用的参数 (newX[, i])

我应该如何使用 apply 以便我的 UDF 逐行评估测试集？

我期待着：

[1] 4 7 10

我提供了一个简化的通用 UDF，因为我需要使用更复杂的 UDF。

Answer 1

您可以通过将函数直接传递给apply

apply(test, 1, function(x) if(x[1] > 0) sum(x) else x[1] - x[2] - x[3])

[1]  4  7 10

如果你想使用你的 UDF，你需要修改它。

testfn = function(mydf){
  if(mydf[1] > 0){y = mydf[1] + mydf[2] + mydf[3]}
  if(mydf[1] < 0){y = mydf[1] - mydf[2] - mydf[3]}
  return(y)
}

apply(test, 1, testfn)