如何在 Matplotlib 中绘制掩码和 NaN 值?

若要在 Matplotlib 中绘制掩码和 NaN 值，我们可以采取以下步骤 -

  • 设置图形大小和调整子图之间及周围的边距。
  • 使用 numpy 创建 x 和 y 数据点。
  • 获取 x2 和 y2 数据点，使得 y > 0.7。
  • 获取掩码 y3 数据点，使得 y > 0.7。
  • 将 y3 掩码为 NaN 值。
  • 使用 plot() 方法绘制 x、y、y2、y3 和 y4。
  • 为绘图添加图例。
  • 设置绘图的标题。
  • 要显示图形，请使用 show() 方法。

示例

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

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

输出

更新于：2021-6-10

3K+ 浏览次数

开启你的 职业生涯

通过完成课程获得认证

开始
广告