Créer des graphiques en barres empilées avec Matplotlib

Introduction

Dans ce laboratoire, nous allons apprendre à créer un graphique en barres empilées à l'aide de la bibliothèque Matplotlib en Python. Nous utiliserons des données sur les pingouins pour créer un graphique en barres empilées qui montre le nombre de pingouins ayant une masse corporelle supérieure à la moyenne.

Conseils sur la VM

Une fois le démarrage de la VM terminé, cliquez dans le coin supérieur gauche pour basculer vers l'onglet Notebook pour accéder à Jupyter Notebook pour la pratique.

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

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

Nous allons commencer par importer les bibliothèques nécessaires, y compris numpy et matplotlib.pyplot.

import matplotlib.pyplot as plt
import numpy as np

Définir les données

Nous allons définir les données que nous utiliserons pour créer le graphique en barres empilées.

species = (
    "Adelie\n $\\mu=$3700.66g",
    "Chinstrap\n $\\mu=$3733.09g",
    "Gentoo\n $\\mu=5076.02g$",
)
weight_counts = {
    "Below": np.array([70, 31, 58]),
    "Above": np.array([82, 37, 66]),
}
width = 0.5

Créer un graphique en barres empilées

Nous allons créer un graphique en barres empilées à l'aide de matplotlib.pyplot.bar et parcourir chaque catégorie de poids pour empiler les barres.

fig, ax = plt.subplots()
bottom = np.zeros(3)

for boolean, weight_count in weight_counts.items():
    p = ax.bar(species, weight_count, width, label=boolean, bottom=bottom)
    bottom += weight_count

ax.set_title("Number of penguins with above average body mass")
ax.legend(loc="upper right")

Afficher le graphique

Nous allons afficher le graphique en barres empilées à l'aide de matplotlib.pyplot.show().

plt.show()

Sommaire

Dans ce laboratoire, nous avons appris à créer un graphique en barres empilées à l'aide de la bibliothèque Matplotlib en Python. Nous avons commencé par importer les bibliothèques nécessaires, défini les données que nous utiliserions pour créer le graphique, puis créé un graphique en barres empilées à l'aide de matplotlib.pyplot.bar. Enfin, nous avons affiché le graphique à l'aide de matplotlib.pyplot.show().

Graphique en barres empilées avec Matplotlib