Ajouter une barre de couleur à des axes insérés

Introduction

Dans ce laboratoire, vous allez apprendre à ajouter une barre de couleur à des axes insérés à l'aide de Matplotlib en Python. Une barre de couleur est une représentation visuelle de la correspondance entre une plage de couleurs et une plage de valeurs numériques. Un axe inséré est un axe plus petit qui est placé à l'intérieur de l'axe plus grand d'un graphique.

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 de notes 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.

Importation des bibliothèques et des données

Tout d'abord, importez les bibliothèques et les données nécessaires qui seront utilisées dans le graphique.

import matplotlib.pyplot as plt
from matplotlib import cbook
from mpl_toolkits.axes_grid1.inset_locator import inset_axes, zoomed_inset_axes

fig, ax = plt.subplots(figsize=[5, 4])

Z = cbook.get_sample_data("axes_grid/bivariate_normal.npy")
extent = (-3, 4, -4, 3)

Configurer le graphique principal

Configurez le graphique principal en ajustant le rapport d'aspect et les limites des axes x et y.

ax.set(aspect=1, xlim=(-15, 15), ylim=(-20, 5))

Créer un axe inséré

Créez un axe inséré à l'aide de la fonction zoomed_inset_axes. Réglez le niveau de zoom et l'emplacement de l'axe inséré dans le graphique principal.

axins = zoomed_inset_axes(ax, zoom=2, loc='upper left')
axins.set(xticks=[], yticks=[])

Ajouter une image à l'axe inséré

Ajoutez une image à l'axe inséré à l'aide de la fonction imshow. Réglez l'étendue et l'origine de l'image.

im = axins.imshow(Z, extent=extent, origin="lower")

Ajouter une barre de couleur

Ajoutez une barre de couleur à l'axe inséré à l'aide de la fonction inset_axes. Réglez la largeur, la hauteur, l'emplacement et la boîte englobante de la barre de couleur.

cax = inset_axes(axins,
                 width="5%",  ## width = 10% of parent_bbox width
                 height="100%",  ## height : 50%
                 loc='lower left',
                 bbox_to_anchor=(1.05, 0., 1, 1),
                 bbox_transform=axins.transAxes,
                 borderpad=0,
                 )
fig.colorbar(im, cax=cax)

Afficher le graphique

Affichez le graphique à l'aide de la fonction show.

plt.show()

Sommaire

Félicitations ! Vous avez appris avec succès à ajouter une barre de couleur à des axes insérés à l'aide de Matplotlib en Python. Il s'agit d'une technique utile pour visualiser les données de manière plus détaillée et informative. N'oubliez pas d'ajuster les paramètres selon vos besoins et vos préférences spécifiques.