промежуточные итоги по категории по нескольким столбцам

Question

Я ищу эффективное решение для добавления промежуточных итогов для каждого столбца в новой строке для каждой категории в столбце 'id'. Я могу добиться желаемого результата с помощью приведенного ниже кода, но этот подход не эффективен для больших наборов данных. Можно ли сделать это, используя таблицы данных?

Спасибо!

data <- data.frame(id = c("a","b","a","b","c","c","c","a","a","b"),
               total = c(1,2,3,4,2,3,4,2,3,4),
               total2 = c(2,3,4,2,3,4,5,6,4,2),
               total3 = c(2,3,4,5,6,3,2,3,4,5))

data_new <- data.frame(id = character(), total = numeric(), total2 = 
numeric(), total3 = numeric())

for (i in unique(data$id)){

subset <- data[data$id == i,]

subtotals <- data.frame(id = i, total = sum(subset$total), total2 = 
sum(subset$total2), total3 = sum(subset$total3))

subset <- rbind(subset,subtotals)

data_new <- rbind(data_new, subset)


}

data_new 

Answer 1

Вот базовое решение R с использованием aggregate. Спасибо @thelatemail за оптимизацию оригинальной версии.

SubTotals = aggregate(data[,2:4], data["id"], sum)
data_new = rbind(data, SubTotals)
data_new = data_new[order(data_new$id),]
data_new
   id total total2 total3
1   a     1      2      2
3   a     3      4      4
8   a     2      6      3
9   a     3      4      4
11  a     9     16     13
2   b     2      3      3
4   b     4      2      5
10  b     4      2      5
12  b    10      7     13
5   c     2      3      6
6   c     3      4      3
7   c     4      5      2
13  c     9     12     11

Answer 2

И для хорошей меры вот решение data.table:

library(data.table)
setDT(data)
rbind(data, data[, lapply(.SD,sum), by=id])[order(id)]
#    id total total2 total3
# 1:  a     1      2      2
# 2:  a     3      4      4
# 3:  a     2      6      3
# 4:  a     3      4      4
# 5:  a     9     16     13
# 6:  b     2      3      3
# 7:  b     4      2      5
# 8:  b     4      2      5
# 9:  b    10      7     13
#10:  c     2      3      6
#11:  c     3      4      3
#12:  c     4      5      2
#13:  c     9     12     11

Сгруппируйте by= переменную id, затем рассчитайте сумму для каждой переменной, отличной от id, через lapply(.SD,sum). Затем rbind это обратно к основному набору, затем order строк на id.

Answer 3

Вот подход в стиле тидиверс:

library(tidyverse)

data_new <- bind_rows(data, 
                      data %>% 
                        group_by(id) %>% 
                        summarise_all(sum)
                      ) %>% 
  arrange(id) %>% 
  print()
#>    id total total2 total3
#> 1   a     1      2      2
#> 2   a     3      4      4
#> 3   a     2      6      3
#> 4   a     3      4      4
#> 5   a     9     16     13
#> 6   b     2      3      3
#> 7   b     4      2      5
#> 8   b     4      2      5
#> 9   b    10      7     13
#> 10  c     2      3      6
#> 11  c     3      4      3
#> 12  c     4      5      2
#> 13  c     9     12     11

^{Создано в 2019-01-10 пакетом представ. (v0.2.1)}