我有一个函数,它在一个列表中返回两个值.这两个值都需要添加到数据中.表中有两个新列.函数的计算代价很高,所以我希望避免计算函数两次.下面是一个例子:
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
有什么建议吗?每次调用myfun时,我可以在一个单独的环境中保存r2
个,我只需要一种方法,通过引用一次添加两列.