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 :

Tracer un tableau 2D Matplotlib en utilisant imshow

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 :

Définir cmap Plot 2D array Matplotlib en utilisant imshow

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 :

Tracer un tableau 2D dans Matplotlib en utilisant pcolormesh

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.