我有一个在列表中返回两个值的函数。两个值都需要添加到 data.table 的两个新列中。函数的评估成本很高,所以我想避免两次计算函数。这是示例:
library(data.table)
example(data.table)
DT
x y v
1: a 1 42
2: a 3 42
3: a 6 42
4: b 1 4
5: b 3 5
6: b 6 6
7: c 1 7
8: c 3 8
9: c 6 9
这是我的功能的一个例子。请记住,我说过它的计算成本很高,除此之外,没有办法从其他给定值中推断出一个返回值(如下例所示):
myfun <- function (y, v)
{
ret1 = y + v
ret2 = y - v
return(list(r1 = ret1, r2 = ret2))
}
这是我在一个语句中添加两列的方法。但是,那个需要调用 myfun 两次:
DT[,new1:=myfun(y,v)$r1][,new2:=myfun(y,v)$r2]
x y v new1 new2
1: a 1 42 43 -41
2: a 3 42 45 -39
3: a 6 42 48 -36
4: b 1 4 5 -3
5: b 3 5 8 -2
6: b 6 6 12 0
7: c 1 7 8 -6
8: c 3 8 11 -5
9: c 6 9 15 -3
关于如何做到这一点的任何建议?r2
每次调用 myfun 时,我都可以保存在单独的环境中,我只需要一种方法来一次通过引用添加两列。