Graphique en barres groupées avec des annotations utilisant Matplotlib

Introduction

Dans ce laboratoire, nous allons apprendre à créer un graphique en barres groupées et à annoter les barres avec des étiquettes à l'aide de Python Matplotlib. Nous utiliserons les données du jeu de données Palmer Penguins pour créer un graphique qui affiche les attributs des pingouins par espèce.

Conseils sur la machine virtuelle

Une fois le démarrage de la machine virtuelle terminé, cliquez dans le coin supérieur gauche pour basculer vers l'onglet Carnet de notes pour accéder au carnet Jupyter pour pratiquer.

Parfois, vous devrez peut-être attendre quelques secondes pour que le carnet Jupyter ait fini de charger. La validation des opérations ne peut pas être automatisée en raison des limitations du carnet Jupyter.

Si vous rencontrez des problèmes pendant l'apprentissage, n'hésitez pas à demander à Labby. Donnez des commentaires après la session, et nous résoudrons rapidement le problème pour vous.

Importation des bibliothèques requises

Nous allons commencer par importer les bibliothèques nécessaires pour travailler avec nos données et créer le graphique.

import matplotlib.pyplot as plt
import numpy as np

Préparer les données

Ensuite, nous allons préparer les données pour notre graphique. Nous avons trois espèces de pingouins et trois attributs, donc nous allons créer un dictionnaire avec les moyennes de chaque attribut par espèce.

species = ("Adelie", "Chinstrap", "Gentoo")
penguin_means = {
    'Bill Depth': (18.35, 18.43, 14.98),
    'Bill Length': (38.79, 48.83, 47.50),
    'Flipper Length': (189.95, 195.82, 217.19),
}

Créer un graphique en barres groupées

Maintenant, nous pouvons créer notre graphique en utilisant la fonction bar de Matplotlib. Nous allons créer une boucle qui parcourt nos attributs et crée un ensemble de barres pour chacun d'entre eux. Nous allons également ajuster la largeur des barres et la position de chaque ensemble de barres.

x = np.arange(len(species))
width = 0.25
multiplier = 0

fig, ax = plt.subplots()

for attribute, measurement in penguin_means.items():
    offset = width * multiplier
    rects = ax.bar(x + offset, measurement, width, label=attribute)
    multiplier += 1

Ajouter des étiquettes aux barres

Nous pouvons ajouter des étiquettes aux barres en utilisant la fonction bar_label de Matplotlib. Cela ajoutera la valeur de chaque barre au-dessus d'elle.

ax.bar_label(rects, padding=3)

Personnaliser le graphique

Nous pouvons personnaliser le graphique en ajoutant des étiquettes, un titre et en ajustant les étiquettes d'échelonnement de l'axe x et la légende. Nous allons également définir la limite de l'axe y pour vous assurer que toutes nos données sont visibles.

ax.set_ylabel('Length (mm)')
ax.set_title('Penguin attributes by species')
ax.set_xticks(x + width, species)
ax.legend(loc='upper left', ncols=3)
ax.set_ylim(0, 250)

Afficher le graphique

Enfin, nous pouvons afficher le graphique en utilisant la fonction show de Matplotlib.

plt.show()

Sommaire

Dans ce laboratoire, nous avons appris à créer un graphique en barres groupées et à annoter les barres avec des étiquettes à l'aide de Python Matplotlib. Nous avons utilisé les données du jeu de données Palmer Penguins pour créer un graphique qui affiche les attributs des pingouins par espèce. Nous avons également appris à personnaliser le graphique en ajoutant des étiquettes, un titre et en ajustant les étiquettes d'échelonnement de l'axe x et la légende.

Graphique en barres groupées avec étiquettes utilisant Matplotlib