繪製遮罩和 NaN 值

有時您需要繪製具有遺失值的數據。

一種可能性是簡單地移除不需要的數據點。透過剩餘數據繪製的線條將是連續的，並且不會指出遺失數據的位置。

如果需要在遺失數據的位置產生線條間隙，則可以使用遮罩陣列或將它們的值設定為 NaN 來表示不需要的點。在 x 或 y 被遮罩的位置將不會繪製標記，如果使用線條繪圖，則會在那裡斷開。

以下範例說明了三種情況

1. 移除點。

2. 遮罩點。

3. 設定為 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()

標籤：繪圖類型：線條 級別：中級