Comment créer un graphique à barres empilées dans Pandas

Vous pouvez utiliser la syntaxe de base suivante pour créer un graphique à barres empilées dans pandas :

df.groupby(['var1', 'var2']).size().unstack().plot(kind='bar', stacked=True)

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Exemple : créer un graphique à barres empilées dans Pandas

Supposons que nous ayons le DataFrame pandas suivant qui contient des informations sur divers joueurs de basket-ball :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   'position': ['G', 'G', 'F', 'F', 'G', 'F', 'F', 'F'],
                   'points': [5, 7, 7, 9, 12, 9, 9, 4]})

#view DataFrame
print(df)

  team position  points
0    A        G       5
1    A        G       7
2    A        F       7
3    A        F       9
4    B        G      12
5    B        F       9
6    B        F       9
7    B        F       4

Nous pouvons utiliser le code suivant pour créer un graphique à barres empilées qui affiche le nombre total de positions , regroupées par équipe :

df.groupby(['team', 'position']).size().unstack().plot(kind='bar', stacked=True)

L’axe des x montre le nom de l’équipe et l’axe des y montre le nombre total de positions pour chaque équipe.

Sur le tableau, nous pouvons voir que l’équipe A a 2 gardes (G) et 2 attaquants (F) tandis que l’équipe B a 1 garde et 3 attaquants.

On peut également utiliser les arguments color et title pour modifier la couleur des barres et ajouter un titre au graphique :

df.groupby(['team', 'position']).size().unstack().plot(kind='bar', stacked=True,
            color=['steelblue','pink'], title='Position Count by Team')

Un titre a été ajouté en haut de l’intrigue et les couleurs des barres ont été changées en bleu acier et rose, comme nous l’avons précisé.

Ressources additionnelles

Les didacticiels suivants expliquent comment créer d’autres graphiques courants en Python :

Comment créer des cartes thermiques en Python
Comment créer une courbe en cloche en Python
Comment créer un graphique Ogive en Python