Créer des légendes pour les tracés Matplotlib

Introduction

Matplotlib est une bibliothèque Python qui permet aux utilisateurs de créer différents types de graphiques et de tracés. Ce tutoriel vous guidera tout au long du processus de création d'une légende dans un tracé Matplotlib.

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

Importez les bibliothèques nécessaires

Avant de commencer, nous devons importer les bibliothèques nécessaires. Dans ce cas, nous allons utiliser la bibliothèque Matplotlib.

import matplotlib.pyplot as plt

Créez une figure et un sous-graphique

Nous devons créer une figure et un sous-graphique pour tracer nos données. Nous allons créer un tracé avec deux sous-graphiques.

fig = plt.figure()

ax = fig.add_subplot(211)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

ax = fig.add_subplot(223)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

Ajoutez une légende au tracé

Nous allons maintenant ajouter une légende au tracé. Il existe deux manières d'ajouter une légende dans Matplotlib. Nous utiliserons les deux méthodes dans cet exemple.

## Méthode 1 : Placer une légende au-dessus du sous-graphique
ax.legend(bbox_to_anchor=(0., 1.02, 1.,.102), loc='lower left',
           ncols=2, mode="expand", borderaxespad=0.)

## Méthode 2 : Placer une légende à droite du sous-graphique
ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left', borderaxespad=0.)

Affichez le tracé

Enfin, nous allons afficher le tracé.

plt.show()

Sommaire

Dans ce tutoriel, nous avons appris à ajouter une légende à un tracé Matplotlib. Nous avons utilisé deux méthodes différentes pour ajouter une légende. La première méthode a placé la légende au-dessus du sous-graphique, tandis que la seconde méthode a placé la légende à droite du sous-graphique.