Ajouter une barre de couleur à un graphique Matplotlib

Beginner

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

Introduction

Matplotlib est une bibliothèque Python utilisée pour la visualisation de données. Dans ce laboratoire, nous allons apprendre à ajouter une barre de couleur à un graphique dans Matplotlib. Les barres de couleur sont utiles pour indiquer la plage de valeurs qu'un mappage de couleurs représente.

Conseils sur la machine virtuelle

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

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.

Importation des bibliothèques nécessaires

Nous allons commencer par importer les bibliothèques nécessaires. Nous utiliserons le module pyplot de Matplotlib, qui fournit une interface pour créer des graphiques.

import matplotlib.pyplot as plt

Créer un graphique

Ensuite, nous allons créer un graphique en utilisant la fonction imshow de Matplotlib. Cette fonction affiche une image sur le graphique. Nous allons également créer une figure avec deux sous-graphiques.

fig, (ax1, ax2) = plt.subplots(1, 2)
fig.subplots_adjust(wspace=0.5)

im1 = ax1.imshow([[1, 2], [3, 4]])

im2 = ax2.imshow([[1, 2], [3, 4]])

Ajouter une barre de couleur au graphique

Maintenant, nous allons ajouter une barre de couleur à chaque sous-graphique en utilisant la fonction make_axes_locatable de Matplotlib. Cette fonction prend un axe existant, l'ajoute à un nouvel AxesDivider et renvoie l'AxesDivider. La méthode append_axes de l'AxesDivider peut ensuite être utilisée pour créer un nouvel axe sur un côté donné ("haut", "droite", "bas" ou "gauche") de l'axe original.

ax1_divider = make_axes_locatable(ax1)
cax1 = ax1_divider.append_axes("right", size="7%", pad="2%")
cb1 = fig.colorbar(im1, cax=cax1)

ax2_divider = make_axes_locatable(ax2)
cax2 = ax2_divider.append_axes("top", size="7%", pad="2%")
cb2 = fig.colorbar(im2, cax=cax2, orientation="horizontal")
cax2.xaxis.set_ticks_position("top")

Afficher le graphique

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

plt.show()

Sommaire

Dans ce laboratoire, nous avons appris à ajouter une barre de couleur à un graphique dans Matplotlib. Nous avons utilisé la fonction make_axes_locatable pour ajouter un axe supplémentaire au graphique et la fonction colorbar pour créer la barre de couleur. Nous avons également appris à changer l'orientation et la position de la barre de couleur.