Créer un diagramme en barres groupées avec Matplotlib

Beginner

This tutorial is from open-source community. Access the source code

Introduction

Dans ce laboratoire, vous allez apprendre à créer un diagramme en barres agrégées avec des unités à l'aide de Matplotlib. Le diagramme en barres affichera la hauteur des tasses par groupe et le choix de boisson. Le tutoriel vous guidera tout au long des étapes de création du graphique à l'aide du code Python.

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 d'adresses 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 votre feedback après la session, et nous résoudrons rapidement le problème pour vous.

Importez les bibliothèques nécessaires

La première étape consiste à importer les bibliothèques nécessaires. Nous allons utiliser NumPy et Matplotlib pour créer le diagramme en barres. Les unités cm et inch sont définies dans le fichier basic_units.py, qui est utilisé pour convertir les unités.

import matplotlib.pyplot as plt
import numpy as np
from basic_units import cm, inch

Définissez les données

L'étape suivante est de définir les données que nous utiliserons pour créer le diagramme en barres. Nous allons créer deux ensembles de données - l'un pour le thé et l'autre pour le café. Chaque ensemble aura cinq valeurs, une pour chaque groupe. La hauteur des tasses sera représentée en centimètres.

N = 5
tea_means = [15*cm, 10*cm, 8*cm, 12*cm, 5*cm]
tea_std = [2*cm, 1*cm, 1*cm, 4*cm, 2*cm]
coffee_means = (14*cm, 19*cm, 7*cm, 5*cm, 10*cm)
coffee_std = (3*cm, 5*cm, 2*cm, 1*cm, 2*cm)

Créez les objets Figure et Axes

L'étape suivante est de créer les objets Figure et Axes. Nous allons utiliser la fonction subplots() pour créer les objets Figure et Axes.

fig, ax = plt.subplots()
ax.yaxis.set_units(inch)

Définissez les paramètres du diagramme en barres

L'étape suivante est de définir les paramètres du diagramme en barres. Nous allons définir les emplacements x pour les groupes, la largeur des barres et les étiquettes pour les graduations de l'axe x.

ind = np.arange(N)    ## les emplacements x pour les groupes
width = 0.35         ## la largeur des barres
ax.set_xticks(ind + width / 2, labels=['G1', 'G2', 'G3', 'G4', 'G5'])

Créez le diagramme en barres

L'étape suivante est de créer le diagramme en barres. Nous allons utiliser la fonction bar() pour créer le graphique. Nous allons créer deux ensembles de barres, l'une pour le thé et l'autre pour le café. Nous ajouterons également des barres d'erreur au graphique.

ax.bar(ind, tea_means, width, bottom=0*cm, yerr=tea_std, label='Tea')
ax.bar(ind + width, coffee_means, width, bottom=0*cm, yerr=coffee_std,
       label='Coffee')

Ajoutez des étiquettes et un titre au graphique

La dernière étape est d'ajouter des étiquettes et un titre au graphique. Nous allons ajouter un titre au graphique, une étiquette pour l'axe x et une légende pour le graphique.

ax.set_title('Cup height by group and beverage choice')
ax.set_xlabel('Group')
ax.legend()
ax.autoscale_view()

Sommaire

Dans ce laboratoire, vous avez appris à créer un diagramme en barres groupées avec des unités à l'aide de Matplotlib. Vous avez appris à importer les bibliothèques nécessaires, à définir les données, à créer les objets Figure et Axes, à définir les paramètres du diagramme en barres, à créer le diagramme en barres et à ajouter des étiquettes et un titre au graphique. Avec ces connaissances, vous pouvez créer vos propres diagrammes en barres à l'aide de Matplotlib.