Comment calculer la moyenne par groupe dans R (avec exemples)



Souvent, vous souhaiterez peut-être calculer la moyenne par groupe dans R. Vous pouvez utiliser trois méthodes pour ce faire :

Méthode 1 : utilisez la base R.

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

Méthode 2 : utilisez le package dplyr().

library(dplyr)

df %>%
  group_by(col_to_group_by) %>%
  summarise_at(vars(col_to_aggregate), list(name = mean))

Méthode 3 : utilisez le package data.table.

library(data.table)

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

Les exemples suivants montrent comment utiliser chacune de ces méthodes dans la pratique.

Méthode 1 : calculer la moyenne par groupe en utilisant la base R

Le code suivant montre comment utiliser la fonction Aggregate() de la base R pour calculer la moyenne des points marqués par équipe dans le bloc de données suivant :

#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    a   5    8
2    a   8    8
3    b  14    9
4    b  18    3
5    b   5    8
6    c   7    7
7    c   7    4

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

  Group.1        x
1       a  6.50000
2       b 12.33333
3       c  7.00000

Méthode 2 : calculer la moyenne par groupe à l’aide de dplyr

Le code suivant montre comment utiliser les fonctions group_by () et summarise_at() du package dplyr pour calculer la moyenne des points marqués par équipe dans le bloc de données suivant :

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 mean points scored by team 
df %>%
  group_by(team) %>%
  summarise_at(vars(pts), list(name = mean))

# A tibble: 3 x 2
  team   name
  <fct> <dbl>
1 a       6.5
2 b      12.3
3 c       7  

Méthode 3 : calculer la moyenne par groupe à l’aide de data.table

Le code suivant montre comment calculer la moyenne des points marqués par équipe dans le bloc de données suivant :

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 mean points scored by team 
df[ ,list(mean=mean(pts)), by=team]

   team     mean
1:    a  6.50000
2:    b 12.33333
3:    c  7.00000

Notez que les trois méthodes renvoient des résultats identiques.

Connexe : Un guide complet de la fonction moyenne dans R

Ressources additionnelles

Comment calculer la somme par groupe dans R
Comment calculer des quantiles par groupe dans R

Ajouter un commentaire

Votre adresse e-mail ne sera pas publiée. Les champs obligatoires sont indiqués avec *