Colorplot di 2D Array Matplotlib

Suraj Joshi 15 febbraio 2024
  1. Traccia array 2D in Matplotlib usando il metodo matplotlib.pyplot.imshow()
  2. Traccia array 2D in Matplotlib usando il metodo matplotlib.pyplot.pcolormesh()
Colorplot di 2D Array Matplotlib

Questo tutorial spiega come possiamo generare grafici colorplot di array 2D usando i metodi matplotlib.pyplot.imshow() e matplotlib.pyplot.pcolormesh() in Python.

Traccia array 2D in Matplotlib usando il metodo matplotlib.pyplot.imshow()

Il metodo matplotlib.pyplot.imshow() accetta un array 2D come input e rende l’array dato come un’immagine raster.

Sintassi di 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() Esempio

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()

Produzione:

Traccia un array 2D Matplotlib usando imshow

Traccia l’array 2D creato utilizzando numpy.random.randint() di dimensione 10*10. Per impostazione predefinita, i valori vengono mappati utilizzando la mappa dei colori viridis.

Possiamo impostare il parametro cmap nel metodo imshow() per cambiare la mappa dei colori.

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()

Produzione:

Imposta cmap Plot 2D array Matplotlib usando imshow

Visualizza il grafico dell’array 2D con la mappa dei colori inferno. Possiamo anche vedere una barra dei colori sul lato destro del grafico, che ci dice quali valori nella matrice sono mappati a quali colori.

Traccia array 2D in Matplotlib usando il metodo matplotlib.pyplot.pcolormesh()

La funzione matplotlib.pyplot.pcolormesh() crea un grafico a pseudocolori in Matplotlib. È simile alla funzione 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()

Produzione:

Traccia un array 2D Matplotlib usando pcolormesh

Traccia l’array 2D creato usando numpy.random.randint() di dimensione 10*10 con mappa dei colori plasma. La barra dei colori a destra rappresenta i colori assegnati a diversi intervalli di valori.

Autore: Suraj Joshi
Suraj Joshi avatar Suraj Joshi avatar

Suraj Joshi is a backend software engineer at Matrice.ai.

LinkedIn