Tracciare valori mascherati e NaN

A volte è necessario tracciare dati con valori mancanti.

Una possibilità è rimuovere semplicemente i punti dati indesiderati. La linea tracciata attraverso i dati rimanenti sarà continua e non indicherà dove si trovano i dati mancanti.

Se è utile avere degli spazi nella riga in cui mancano i dati, allora i punti indesiderati possono essere indicati utilizzando un array mascherato o impostando i loro valori su NaN. Non verrà tracciato alcun indicatore dove x o y sono mascherati e, se si traccia con una linea, verrà interrotta in quel punto.

L'esempio seguente illustra i tre casi:

1. Rimozione di punti.

2. Punti di mascheramento.

3. Impostazione su NaN.

import matplotlib.pyplot as plt
import numpy as np


x = np.linspace(-np.pi/2, np.pi/2, 31)
y = np.cos(x)**3

# 1) remove points where y > 0.7
x2 = x[y <= 0.7]
y2 = y[y <= 0.7]

# 2) mask points where y > 0.7
y3 = np.ma.masked_where(y > 0.7, y)

# 3) set to NaN where y > 0.7
y4 = y.copy()
y4[y3 > 0.7] = np.nan

plt.plot(x*0.1, y, 'o-', color='lightgrey', label='No mask')
plt.plot(x2*0.4, y2, 'o-', label='Points removed')
plt.plot(x*0.7, y3, 'o-', label='Masked values')
plt.plot(x*1.0, y4, 'o-', label='NaN values')
plt.legend()
plt.title('Masked and NaN data')
plt.show()