绘制掩码值和 NaN

有时您需要绘制具有缺失值的数据。

一种可能性是简单地删除不需要的数据点。通过剩余数据绘制的线将是连续的,并且不指示缺失数据的位置。

如果在缺少数据的行中有间隙是有用的,则可以使用掩码数组或通过将它们的值设置为 NaN 来指示不需要的点。在 x 或 y 被遮盖的地方不会绘制任何标记,如果用线绘制,它将在那里被破坏。

以下示例说明了这三种情况:

  1. 删除点。

  2. 掩蔽点。

  3. 设置为 NaN。

掩码和 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()

由 Sphinx-Gallery 生成的画廊