Como calcular a soma acumulada por grupo em r

Você pode usar os seguintes métodos para calcular uma soma cumulativa por grupo em R:

Método 1: Use Base R

 df$cum_sum <- ave(df$values_var, df$group_var, FUN=cumsum)

Método 2: use dplyr

 library (dplyr)

df %>% group_by(group_var) %>% mutate(cum_sum = cumsum(values_var))

Método 3: use data.table

 library (data.table)

setDT(df)[, cum_sum := cumsum(values_var), group_var] 

Os exemplos a seguir mostram como usar cada método na prática com o seguinte quadro de dados em R:

 #create data frame
df <- data. frame (store=rep(c(' A ', ' B ', ' C '), each= 4 ),
                 sales=c(3, 4, 4, 2, 5, 8, 9, 7, 6, 8, 3, 2))

#view data frame
df

   blind sales
1 to 3
2 to 4
3 to 4
4 to 2
5 B 5
6 B 8
7 B 9
8 B 7
9 C 6
10 C 8
11 C 3
12 C 2

Exemplo 1: calcule a soma acumulada por grupo usando a base R

O código a seguir mostra como usar a função ave() do banco de dados R para calcular a soma acumulada de vendas , agrupadas por loja :

 #add column to show cumulative sales by store
df$cum_sales <- ave(df$sales, df$store, FUN=cumsum)

#view updated data frame
df

   store sales cum_sales
1 to 3 3
2 to 4 7
3 to 4 11
4 to 2 13
5 B 5 5
6 B 8 13
7 B 9 22
8 B 7 29
9 C 6 6
10 C 8 14
11 C 3 17
12 C 2 19

A nova coluna chamada cum_sales exibe a soma acumulada das vendas , agrupadas por loja .

Exemplo 2: Calcule a soma cumulativa por grupo usando dplyr

O código a seguir demonstra como usar diversas funções do pacote dplyr em R para calcular a soma acumulada de vendas , agrupadas por loja :

 library (dplyr)

#add column to show cumulative sales by store
df %>% group_by(store) %>% mutate(cum_sales = cumsum(sales))

#view updated data frame
df

# A tibble: 12 x 3
# Groups: store [3]
   store sales cum_sales
         
 1 to 3 3
 2 to 4 7
 3 to 4 11
 4 to 2 13
 5 B 5 5
 6 B 8 13
 7 B 9 22
 8 B 7 29
 9 C 6 6
10 C 8 14
11 C 3 17
12 C 2 19

A nova coluna chamada cum_sales exibe a soma acumulada das vendas , agrupadas por loja .

Exemplo 3: Calcular soma cumulativa por grupo usando data.table

O código a seguir demonstra como usar diversas funções do pacote data.table em R para calcular a soma acumulada de vendas , agrupadas por loja :

 library (data.table)

#add column to show cumulative sales by store
setDT(df)[, cum_sales := cumsum(sales), store] 

#view updated data frame
df

    store sales cum_sales
 1: A 3 3
 2: A 4 7
 3: A 4 11
 4: A 2 13
 5: B 5 5
 6: B 8 13
 7: B 9 22
 8: B 7 29
 9: C 6 6
10: C 8 14
11: C 3 17
12: C 2 19

A nova coluna chamada cum_sales exibe a soma acumulada das vendas , agrupadas por loja .

Nota : Todos os três métodos produzem o mesmo resultado. No entanto, os métodos dplyr e data.table tenderão a ser mais rápidos ao trabalhar com quadros de dados extremamente grandes.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outros cálculos comuns em R:

Como calcular soma por grupo em R
Como calcular a média por grupo em R
Como calcular o desvio padrão por grupo em R