R语言data.table：根据列a长度计算嵌套列表列b的元素和

我来帮你搞定这个data.table的计算问题～先看看你遇到的问题：你想根据每行a的数值，取出b列表里的前a个元素求和，但直接用dt[, c:= sum(b[[1]][1:a])]得到的结果不对还出警告，这是因为data.table的j表达式默认是对整列操作，不是逐行处理的。

问题根源 #

你的原代码里，a是长度为4的向量（1:4），当你写b[[1]][1:a]时，R会把1:a解析成c(1:1, 1:2, 1:3, 1:4)，也就是c(1,1,2,1,2,3,1,2,3,4)，然后sum会把这些索引对应的元素全部加起来，得到一个单一值，再赋值给整个c列，这显然不是你要的逐行计算效果。

正确实现方式 #

这里给你两种靠谱的方法：

方法1：用sapply/mapply逐行映射

因为你的b列每个元素都是同一个列表1:6，可以直接用sapply遍历a的每个值，计算对应前n项和：

require(data.table)
dt <- data.table(a = 1:4)
dt[, b:= .(list(1:6))]

# 用sapply实现
dt[, c := sapply(a, function(x) sum(b[[1]][1:x]))]

如果b列每行的列表不一样，就用mapply同时遍历a和b：

dt[, c := mapply(function(x, lst) sum(lst[1:x]), a, b)]

方法2：用by = .I逐行处理

data.table里可以用by = .I来强制逐行计算，这样每行的a都是单个值，索引就不会出错了：

dt[, c := sum(b[[1]][1:a]), by = .I]

验证结果 #

运行上面的代码后，dt的c列会得到：

a         b  c
1: 1 1,2,3,4,5,6  1
2: 2 1,2,3,4,5,6  3
3: 3 1,2,3,4,5,6  6
4: 4 1,2,3,4,5,6 10

完全符合预期～

内容的提问来源于stack exchange，提问作者PavoDive