如何在 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()输出

广告
数据结构
网络
RDBMS
操作系统
Java
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP