Alignement de grille d'images dans Matplotlib

Introduction

Dans ce tutoriel, nous allons apprendre à utiliser la fonction ImageGrid du module mpl_toolkits.axes_grid1 dans Matplotlib. Nous allons créer une grille 2x2 d'images et les aligner à l'aide de la fonction ImageGrid.

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 à 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 et créez des tableaux d'images

Nous commençons par importer les bibliothèques nécessaires et en créant quatre tableaux d'images de 10x10 à l'aide des fonctions arange et reshape de la bibliothèque NumPy.

import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.axes_grid1 import ImageGrid

im1 = np.arange(100).reshape((10, 10))
im2 = im1.T
im3 = np.flipud(im1)
im4 = np.fliplr(im2)

Créez une figure et un objet ImageGrid

Ensuite, nous créons un objet figure à l'aide de la fonction plt.figure et passons l'argument figsize pour définir la taille de la figure. Nous créons ensuite un objet ImageGrid à l'aide de la fonction ImageGrid et passons le figure, 111 comme argument pour le sous-graphique, (2, 2) comme argument nrows_ncols pour créer une grille 2x2 d'axes et 0.1 comme argument axes_pad pour définir l'espacement entre les axes.

fig = plt.figure(figsize=(4., 4.))
grid = ImageGrid(fig, 111, nrows_ncols=(2, 2), axes_pad=0.1)

Itérez sur la grille et tracez les images

Ensuite, nous itérons sur l'objet grid à l'aide de la fonction zip pour itérer à la fois sur les axes et les tableaux d'images. Nous traçons chaque image sur son axe correspondant à l'aide de la fonction imshow.

for ax, im in zip(grid, [im1, im2, im3, im4]):
    ax.imshow(im)

Affichez le tracé

Enfin, nous affichons le tracé à l'aide de la fonction plt.show.

plt.show()

Sommaire

Dans ce tutoriel, nous avons appris à utiliser la fonction ImageGrid du module mpl_toolkits.axes_grid1 dans Matplotlib pour aligner plusieurs images dans une grille. Nous avons créé une grille 2x2 d'images et les avons tracées à l'aide de la fonction imshow.