Création d'une légende avec des étiquettes prédéfinies

Introduction

Dans la visualisation de données, une légende est la clé pour interpréter les éléments visuels d'un graphique. Elle aide le spectateur à comprendre les données et la signification de la représentation visuelle. Matplotlib est une bibliothèque Python populaire pour créer des visualisations de données, y compris des graphiques avec légendes. Dans ce tutoriel, nous allons apprendre à créer une légende avec des étiquettes prédéfinies dans 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 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 requises, qui incluent Matplotlib et NumPy. Nous utilisons NumPy pour générer des données factices pour notre graphique.

import matplotlib.pyplot as plt
import numpy as np

Génération des données

Ensuite, nous allons générer des données factices pour les utiliser dans notre graphique. Nous allons créer deux tableaux, a et b, à l'aide de la fonction arange de NumPy. Nous calculons ensuite deux tableaux supplémentaires, c et d, en utilisant la fonction exp pour calculer l'exponentielle de a et d comme l'inverse de c.

## Make some fake data.
a = b = np.arange(0, 3,.02)
c = np.exp(a)
d = c[::-1]

Création du graphique

Maintenant, nous sommes prêts à créer notre graphique. Nous allons utiliser la fonction plot de Matplotlib pour tracer trois lignes sur le même graphique, chacune avec une étiquette prédéfinie. Nous utiliserons le paramètre label pour attribuer les étiquettes à chaque ligne.

## Create plots with pre-defined labels.
fig, ax = plt.subplots()
ax.plot(a, c, 'k--', label='Model length')
ax.plot(a, d, 'k:', label='Data length')
ax.plot(a, c + d, 'k', label='Total message length')

Ajout de la légende

Pour ajouter la légende à notre graphique, nous utilisons la fonction legend de Matplotlib. Nous passons le paramètre loc pour spécifier l'emplacement de la légende et le paramètre shadow pour ajouter un effet d'ombre à la légende. Nous utilisons également le paramètre fontsize pour définir la taille de police de la légende.

legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')

Stylisation de la légende

Enfin, nous pouvons styliser la légende pour la rendre plus visuellement attrayante. Nous utilisons la fonction get_frame pour obtenir le cadre de la légende, puis la fonction set_facecolor pour définir la couleur d'arrière-plan du cadre.

## Put a nicer background color on the legend.
legend.get_frame().set_facecolor('C0')

Affichage du graphique

Nous pouvons maintenant afficher le graphique à l'aide de la fonction show de Matplotlib.

plt.show()

Sommaire

Dans ce tutoriel, nous avons appris à créer une légende avec des étiquettes prédéfinies dans Matplotlib. Nous avons utilisé la fonction plot pour tracer trois lignes sur le même graphique et le paramètre label pour attribuer des étiquettes à chaque ligne. Nous avons ensuite utilisé la fonction legend pour ajouter la légende au graphique et stylisé la légende pour la rendre plus visuellement attrayante. En suivant ces étapes, vous pouvez créer des légendes pour vos propres graphiques dans Matplotlib.