Colorplot d'un tableau 2D en Matplotlib

1. Tracer un tableau 2D dans Matplotlib en utilisant la méthode matplotlib.pyplot.imshow()
2. Tracer un tableau en 2D dans Matplotlib en utilisant la méthode matplotlib.pyplot.pcolormesh()

Ce tutoriel explique comment générer des tracés en couleur de tableaux 2D en utilisant les méthodes matplotlib.pyplot.imshow() et matplotlib.pyplot.pcolormesh() en Python.

Tracer un tableau 2D dans Matplotlib en utilisant la méthode matplotlib.pyplot.imshow()

La méthode matplotlib.pyplot.imshow() prend un tableau 2D comme entrée et rend le tableau donné sous forme d’image matricielle.

Syntaxe de la méthode matplotlib.pyplot.imshow()

matplotlib.pyplot.imshow(X,
                         cmap=None,
                         norm=None,
                         aspect=None,
                         interpolation=None,
                         alpha=None,
                         vmin=None,
                         vmax=None,
                         origin=None,
                         extent=None, *,
                         filternorm=True,
                         filterrad=4.0,
                         resample=None,
                         url=None,
                         data=None,
                         **kwargs)

matplotlib.pyplot.imshow() Exemple

import numpy as np
import matplotlib.pyplot as plt

X = np.random.randint(256, size=(10, 10))

fig = plt.figure(figsize=(8, 6))
plt.imshow(X)
plt.title("Plot 2D array")
plt.show()

Production :

Il trace le tableau 2D créé à l’aide de la fonction numpy.random.randint() de taille 10*10. Par défaut, les valeurs sont représentées à l’aide de la carte de couleurs viridis.

Nous pouvons définir le paramètre cmap dans la méthode imshow pour modifier la carte de couleurs.

import numpy as np
import matplotlib.pyplot as plt

X = np.random.randint(256, size=(10, 10))

fig = plt.figure(figsize=(8, 6))
plt.imshow(X, cmap="inferno")
plt.title("Plot 2D array")
plt.colorbar()
plt.show()

Production :

Il affiche le tracé du tableau en 2D avec la carte de couleur inferno. Nous pouvons également voir une barre de couleur sur le côté droit du graphique, qui nous indique quelles valeurs du tableau sont associées à quelles couleurs.

Tracer un tableau en 2D dans Matplotlib en utilisant la méthode matplotlib.pyplot.pcolormesh()

La fonction matplotlib.pyplot.pcolormesh() crée un tracé pseudo-couleur dans Matplotlib. Elle est similaire à la fonction matplotlib.pyplot.pcolor().

import numpy as np
import matplotlib.pyplot as plt

X = np.random.randint(256, size=(10, 10))

fig = plt.figure(figsize=(8, 6))
plt.pcolormesh(X, cmap="plasma")
plt.title("Plot 2D array")
plt.colorbar()
plt.show()

Production :

Il trace le tableau 2D créé à l’aide de la fonction numpy.random.randint() de taille 10*10 avec une carte de couleurs plasma. La barre de couleur à droite représente les couleurs attribuées aux différentes plages de valeurs.