我有一个data.table名字dt,如下所示:
ID V1 V2 V3 V4 V5 time color
1 F T F F T 1 red
1 F T T F T 2 red
2 T F T F F 1 blue
3 F F F F F 2 green
3 T T T F T 3 purple
Run Code Online (Sandbox Code Playgroud)
实际上,dim(dt) = [1321221 123].现在我知道一般来说,真和假分别在R中存储为1和0.我也有一个数组l,虽然看起来像
V1 V2 V3 V4 V5
1 2 1 3 4
Run Code Online (Sandbox Code Playgroud)
这些是分配给的权重V1,V2,V3,V4,V5.我想将这些权重乘以真值(因为它们的数值为1)并在每行中添加,就像我们可以用矩阵一样.输出应该是这样的
ID Total time color
1 6 1 red
1 7 2 red
2 2 1 blue
3 0 2 green
3 8 3 purple
Run Code Online (Sandbox Code Playgroud)
现在请记住,这实际上是真实数据的一小部分,所以我正在寻找一个快速的解决方案,最好是一个data.table解决方案,这样我就可以提高我的data.table技能.
dt[, .(ID, Reduce(`+`, Map(`*`, .SD, l)), time, color), .SDcols = V1:V5]
# ID V2 time color
#1: 1 6 1 red
#2: 1 7 2 red
#3: 2 2 1 blue
#4: 3 0 2 green
#5: 3 8 3 purple
# or using matrix product
dt[, .(ID, as.matrix(.SD) %*% t(l), time, color), .SDcols = V1:V5]
Run Code Online (Sandbox Code Playgroud)