Pandas : comment utiliser groupby() avec size()

Vous pouvez utiliser les méthodes suivantes avec les fonctions groupby() et size() dans pandas pour compter le nombre d’occurrences par groupe :

Méthode 1 : compter les occurrences regroupées par une variable

df.groupby('var1').size()

Méthode 2 : compter les occurrences regroupées par plusieurs variables

df.groupby(['var1', 'var2']).size()

Méthode 3 : compter les occurrences regroupées par plusieurs variables et trier par nombre

df.groupby(['var1', 'var2']).size().sort_values(ascending=False)

Les exemples suivants montrent comment utiliser chaque méthode en pratique avec le DataFrame pandas suivant :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'A', 'A', 'A', 'A', 'B', 'B', 'B', 'B', 'B'],
                   'position': ['G', 'G', 'F', 'F', 'F', 'G', 'G', 'G', 'G', 'F'],
                   'points': [15, 22, 24, 25, 20, 35, 34, 19, 14, 12]})

#view DataFrame
print(df)

  team position  points
0    A        G      15
1    A        G      22
2    A        F      24
3    A        F      25
4    A        F      20
5    B        G      35
6    B        G      34
7    B        G      19
8    B        G      14
9    B        F      12

Exemple 1 : compter les occurrences regroupées par une variable

Le code suivant montre comment utiliser les fonctions groupby() et size() pour compter les occurrences de valeurs dans la colonne team :

#count occurrences of each value in team column
df.groupby('team').size()

team
A    5
B    5
dtype: int64

À partir du résultat, nous pouvons voir que les valeurs A et B apparaissent toutes deux 5 fois dans la colonne équipe .

Exemple 2 : compter les occurrences regroupées par plusieurs variables

Le code suivant montre comment utiliser les fonctions groupby() et size() pour compter les occurrences de valeurs pour chaque combinaison de valeurs dans les colonnes team et position :

#count occurrences of values for each combination of team and position
df.groupby(['team', 'position']).size()

team  position
A     F           3
      G           2
B     F           1
      G           4
dtype: int64

À partir du résultat, nous pouvons voir :

• L’équipe A et la position F apparaissent 3 fois.
• L’équipe A et la position G apparaissent 2 fois.

Et ainsi de suite.

Exemple 3 : compter les occurrences regroupées par plusieurs variables et trier

Le code suivant montre comment utiliser les fonctions groupby() et size() pour compter les occurrences de valeurs pour chaque combinaison de valeurs dans les colonnes team et position , puis trier par nombre :

#count occurrences for each combination of team and position and sort
df.groupby(['team', 'position']).size().sort_values(ascending=False)

team  position
B     G           4
A     F           3
      G           2
B     F           1
dtype: int64

La sortie affiche le nombre de chaque combinaison de valeurs d’ équipe et de position , triées par nombre par ordre décroissant.

Remarque : Pour trier par nombre par ordre croissant, supprimez simplement ascending=False dans la fonction sort_values() .

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes dans les pandas :

Comment compter les valeurs uniques à l’aide de Pandas GroupBy
Comment appliquer une fonction à Pandas Groupby
Comment créer un tracé à barres à partir de Pandas GroupBy