Computa la somma cumulativa per i livelli di una variabile di nidificazione
Domanda
Spero di calcolare una variabile di somma cumulativa, ma calcolata all'interno dei livelli di una data variabile di nidificazione.Ecco alcuni dati di esempio:
data <- data.frame(cbind(ids=c(rep(1,6),rep(2,4),rep(3,3)), values=c(1,5,2,7,3,5,1,6,2,4,1,6,3)))
.
Voglio un nuovo vettore che sarebbe simile a questo:
data$cumsum <- c(1,6,8,15,18,23,1,7,9,13,1,7,10)
.
Con un prodotto finale come questo:
> data
ids values cumsum
1 1 1 1
2 1 5 6
3 1 2 8
4 1 7 15
5 1 3 18
6 1 5 23
7 2 1 1
8 2 6 7
9 2 2 9
10 2 4 13
11 3 1 1
12 3 6 7
13 3 3 10
.
Grazie!
Soluzione
È possibile utilizzare ave
:
transform(data, cumsum = ave(values, ids, FUN = cumsum))
data
# ids values cumsum
#1 1 1 1
#2 1 5 6
#3 1 2 8
#4 1 7 15
#5 1 3 18
#6 1 5 23
#7 2 1 1
#8 2 6 7
#9 2 2 9
#10 2 4 13
#11 3 1 1
#12 3 6 7
#13 3 3 10
.
o usando dplyr
:
library(dplyr)
data %>% group_by(ids) %>% mutate(cumsum = cumsum(values))
. Altri suggerimenti
dovrebbe essere più veloce con data.table
per set di dati di grandi dimensioni
library(data.table)
setDT(data)[, Cumsum:=cumsum(values), by=ids]
data
# ids values Cumsum
# 1: 1 1 1
# 2: 1 5 6
# 3: 1 2 8
# 4: 1 7 15
# 5: 1 3 18
# 6: 1 5 23
# 7: 2 1 1
# 8: 2 6 7
# 9: 2 2 9
# 10: 2 4 13
# 11: 3 1 1
# 12: 3 6 7
# 13: 3 3 10
. Autorizzato sotto: CC-BY-SA insieme a attribuzione
Non affiliato a StackOverflow