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)

R代码：使用带有多个参数的UDF进行应用功能

1 个答案: