Накопленная сумма по столбцам вместо строк
У меня есть data.table
dt
следующее.
df <- data.frame(t1 = rep(0,5), t3 = c(12, 5, 8,9, 5), t7= c(25, 48, 7, 9, 14))
dt <- setDT(df)
dt
t1 t3 t7
1: 0 12 25
2: 0 5 48
3: 0 8 7
4: 0 9 9
5: 0 5 14
Я хочу получить совокупные суммы по столбцам. Я только получаю это через ряды. Как это сделать в data.table
,
dt[, 1:3 := cumsum(dt)]
dt
t1 t3 t7
1: 0 12 25
2: 0 17 73
3: 0 25 80
4: 0 34 89
5: 0 39 103
Желаемый результат выглядит следующим образом:
dt
t1 t3 t7
1: 0 12 37
2: 0 5 53
3: 0 8 15
4: 0 9 18
5: 0 5 19
2 ответа
Решение
Еще один вариант использования Reduce
с accumulate=TRUE
:
dt[, names(dt) := Reduce(`+`, dt, accumulate = TRUE)]
dt
# t1 t3 t7
#1: 0 12 37
#2: 0 5 53
#3: 0 8 15
#4: 0 9 18
#5: 0 5 19
Если нам нужно сделать это по строке, то один вариант - это группа по строке, unlist
, получить cumsum
, преобразовать в list
и назначить его столбцам
dt[, (1:3) := as.list(cumsum(unlist(.SD))), 1:nrow(dt)]
dt
# t1 t3 t7
#1: 0 12 37
#2: 0 5 53
#3: 0 8 15
#4: 0 9 18
#5: 0 5 19
Или другой вариант rowCumsums
от matrixStats
, который может быть применен к matrix
library(matrixStats)
dt[, (1:3) := as.data.table(rowCumsums(as.matrix(.SD)))]