Jak obliczyć sumę według grupy w r (z przykładami)


Często możesz chcieć obliczyć sumę na grupę w R. Możesz w tym celu skorzystać z trzech metod:

Metoda 1: Użyj zasady R.

 aggregate(df$col_to_aggregate, list(df$col_to_group_by), FUN= sum )

Metoda 2: Użyj pakietu dplyr().

 library (dplyr)

df %>%
  group_by (col_to_group_by) %>%
  summarize (Freq = sum (col_to_aggregate))

Metoda 3: Użyj pakietu data.table.

 library (data.table)

dt[ ,list(sum= sum (col_to_aggregate)), by=col_to_group_by]

Poniższe przykłady pokazują, jak zastosować każdą z tych metod w praktyce.

Metoda 1: Oblicz sumę na grupę, korzystając z podstawy R

Poniższy kod pokazuje, jak używać funkcji Aggregate() bazy danych R do obliczenia sumy punktów zdobytych przez drużynę w następującej ramce danych:

 #create data frame
df <- data.frame(team=c('a', 'a', 'b', 'b', 'b', 'c', 'c'),
                 pts=c(5, 8, 14, 18, 5, 7, 7),
                 rebs=c(8, 8, 9, 3, 8, 7, 4))

#view data frame
df

  team pts rebs
1 to 5 8
2 to 8 8
3 b 14 9
4 b 18 3
5 b 5 8
6 c 7 7
7 c 7 4

#find sum of points scored by team
aggregate(df$pts, list(df$team), FUN= sum )

  Group.1 x
1 to 13
2 b 37
3 v 14

Metoda 2: Oblicz sumę według grupy za pomocą dplyr

Poniższy kod pokazuje, jak używać funkcji group_by() i summarise() pakietu dplyr do obliczenia sumy punktów zdobytych przez zespół w następującej ramce danych:

 library (dplyr)

#create data frame
df <- data.frame(team=c('a', 'a', 'b', 'b', 'b', 'c', 'c'),
                 pts=c(5, 8, 14, 18, 5, 7, 7),
                 rebs=c(8, 8, 9, 3, 8, 7, 4))

#find sum of points scored by team
df %>%
group_by (team) %>%
summarize (Freq = sum (pts))

# A tibble: 3 x 2
  team Freq
  <chr> <dbl>
1 to 13
2 b 37
3 v 14

Metoda 3: Oblicz sumę według grupy za pomocą tabeli danych

Poniższy kod pokazuje, jak wykorzystać pakiet data.table do obliczenia sumy punktów zdobytych przez drużynę w poniższej ramce danych:

 library (data.table)

#create data frame
df <- data.frame(team=c('a', 'a', 'b', 'b', 'b', 'c', 'c'),
                 pts=c(5, 8, 14, 18, 5, 7, 7),
                 rebs=c(8, 8, 9, 3, 8, 7, 4))

#convert data frame to data table 
setDT(df)

#find sum of points scored by team
df[,list(sum= sum (pts)), by=team]

   team sum
1:a 13
2:b37
3:c14

Należy pamiętać, że wszystkie trzy metody zwracają identyczne wyniki.

Uwaga: jeśli masz bardzo duży zbiór danych, metoda data.table będzie działać najszybciej z trzech wymienionych tutaj metod.

Dodatkowe zasoby

Jak obliczyć średnią na grupę w R
Jak obliczyć kwantyle według grupy w R

Dodaj komentarz

Twój adres e-mail nie zostanie opublikowany. Wymagane pola są oznaczone *