Personnalisation des sous-graphiques avec GridSpec

Introduction

Dans ce tutoriel, nous allons apprendre à utiliser GridSpec pour générer des sous-graphiques, contrôler les tailles relatives des sous-graphiques avec width_ratios et height_ratios, et l'espacement autour et entre les sous-graphiques à l'aide des paramètres de sous-graphique (left, right, bottom, top, wspace et hspace).

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'étude 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.

Importation des bibliothèques

Nous commençons par importer les bibliothèques nécessaires, qui sont matplotlib.pyplot et GridSpec.

import matplotlib.pyplot as plt
from matplotlib.gridspec import GridSpec

Générer des sous-graphiques avec GridSpec

Dans cette étape, nous allons utiliser GridSpec pour générer des sous-graphiques. Nous allons créer une figure avec 2 lignes et 2 colonnes. Nous spécifierons également les width_ratios et les height_ratios pour contrôler les tailles relatives des sous-graphiques.

fig = plt.figure()
gs = GridSpec(2, 2, width_ratios=[1, 2], height_ratios=[4, 1])
ax1 = fig.add_subplot(gs[0])
ax2 = fig.add_subplot(gs[1])
ax3 = fig.add_subplot(gs[2])
ax4 = fig.add_subplot(gs[3])

Contrôler l'espacement autour et entre les sous-graphiques

Dans cette étape, nous allons utiliser GridSpec pour contrôler l'espacement autour et entre les sous-graphiques. Nous allons créer une figure avec 2 gridspec, chacun avec 3 lignes et 3 colonnes. Nous spécifierons les paramètres left, right, bottom, top, wspace et hspace pour contrôler l'espacement.

fig = plt.figure()
gs1 = GridSpec(3, 3, left=0.05, right=0.48, wspace=0.05)
ax1 = fig.add_subplot(gs1[:-1, :])
ax2 = fig.add_subplot(gs1[-1, :-1])
ax3 = fig.add_subplot(gs1[-1, -1])

gs2 = GridSpec(3, 3, left=0.55, right=0.98, hspace=0.05)
ax4 = fig.add_subplot(gs2[:, :-1])
ax5 = fig.add_subplot(gs2[:-1, -1])
ax6 = fig.add_subplot(gs2[-1, -1])

Annoter les axes

Dans cette étape, nous allons annoter les axes avec leurs numéros de sous-graphique correspondants.

def annotate_axes(fig):
    for i, ax in enumerate(fig.axes):
        ax.text(0.5, 0.5, "ax%d" % (i+1), va="center", ha="center")
        ax.tick_params(labelbottom=False, labelleft=False)

annotate_axes(fig)

Afficher les graphiques

Dans cette étape, nous allons afficher les graphiques.

plt.show()

Sommaire

Dans ce tutoriel, nous avons appris à utiliser GridSpec pour générer des sous-graphiques et contrôler l'espacement autour et entre les sous-graphiques. Nous avons également appris à annoter les axes avec leurs numéros de sous-graphique correspondants. Ces compétences seront utiles pour créer des graphiques et des visualisations complexes.