Come creare un diagramma di contorno in matplotlib

Un diagramma di contorno è un tipo di diagramma che ci consente di visualizzare dati tridimensionali in due dimensioni utilizzando i contorni.

È possibile creare un diagramma di contorno in Matplotlib utilizzando le due funzioni seguenti:

• matplotlib.pyplot.contour() – Crea grafici di contorno.
• matplotlib.pyplot.contourf() – Crea grafici di contorno riempiti.

I seguenti esempi mostrano come utilizzare queste due funzioni nella pratica.

Esempio 1: tracciato del contorno in Matplotlib

Supponiamo di avere i seguenti dati in Python:

 import numpy as np

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = np.sin(X*2+Y)*3 + np.cos(Y+5)

Possiamo utilizzare il seguente codice per creare un diagramma di contorno per i dati:

 import matplotlib.pyplot as plt

plt. outline (X, Y, Z, colors=' black ')

Quando viene utilizzato un solo colore per il grafico, le linee tratteggiate rappresentano valori negativi e le linee continue rappresentano valori positivi.

Un’alternativa è specificare una mappa colori utilizzando l’argomento cmap . Possiamo anche specificare più righe da utilizzare nella trama con l’argomento livelli :

 plt. contour (X, Y, Z, levels= 30 , cmap=' Reds ') 

Abbiamo scelto di utilizzare la cmap “Rossi” ma puoi trovare un elenco completo delle opzioni della tavolozza dei colori nella pagina della documentazione di Matplotlib .

Esempio 2: tracciato del contorno riempito in Matplotlib

Un tracciato di contorno riempito è simile a un tracciato di contorno, tranne per il fatto che gli spazi tra le linee sono riempiti.

Il codice seguente mostra come utilizzare la funzione contourf() per creare un diagramma di contorno riempito per gli stessi dati utilizzati nell’esempio precedente:

 plt. contourf (X, Y, Z, cmap=' Reds ') 

Possiamo anche usare la funzione colorbar() per aggiungere una barra di colore etichettata accanto alla trama:

 plt. contourf (X, Y, Z, cmap=' Reds ')
plt. colorbar () 

Puoi trovare altri tutorial su Matplotlib qui .