Як обчислити суму за групою в r (з прикладами)

Часто вам може знадобитися обчислити суму для кожної групи в R. Для цього можна використати три методи:

Спосіб 1. Використовуйте базу R.

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

Спосіб 2. Використовуйте пакет dplyr().

 library (dplyr)

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

Спосіб 3. Використовуйте пакет data.table.

 library (data.table)

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

Наступні приклади показують, як використовувати кожен із цих методів на практиці.

Спосіб 1: обчисліть суму на групу за основою R

У наступному коді показано, як використовувати функцію Aggregate() бази даних R для обчислення суми очок, набраних командою, у такому кадрі даних:

 #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

Спосіб 2: обчисліть суму за групою за допомогою dplyr

У наступному коді показано, як використовувати функції group_by() і summarise() пакета dplyr для обчислення суми очок, набраних командою, у такому фреймі даних:

 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

Спосіб 3: обчисліть суму за групою за допомогою data.table

У наступному коді показано, як використовувати пакет data.table для обчислення суми очок, набраних командою, у такому фреймі даних:

 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

Зауважте, що всі три методи повертають ідентичні результати.

Примітка. Якщо у вас надзвичайно великий набір даних, метод data.table працюватиме найшвидше з трьох перелічених тут методів.

Додаткові ресурси

Як обчислити середнє значення на групу в R
Як обчислити квантилі за групою в R