简介
Matplotlib 是 Python 中的一个数据可视化库。它是创建各种图形和图表的强大工具。Matplotlib 的特性之一是能够根据 y 值绘制不同颜色的线条。本实验将演示如何使用掩码数组根据 y 值绘制不同颜色的线条。
虚拟机使用提示
虚拟机启动完成后，点击左上角切换到“笔记本”标签页，以访问 Jupyter Notebook 进行练习。
有时，你可能需要等待几秒钟让 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制，操作验证无法自动化。
如果你在学习过程中遇到问题，随时向 Labby 提问。课程结束后提供反馈，我们会立即为你解决问题。
导入所需库
在这一步中，我们将导入本实验所需的库。
import matplotlib.pyplot as plt
import numpy as np
创建数据
在这一步中，我们将为绘图创建数据。我们将创建一个t值的数组和一个s值的数组。
t = np.arange(0.0, 2.0, 0.01)
s = np.sin(2 * np.pi * t)
创建掩码数组
在这一步中，我们将创建三个掩码数组：一个用于大于某个阈值的值，一个用于小于某个阈值的值，还有一个用于介于两个阈值之间的值。
upper = 0.77
lower = -0.77
supper = np.ma.masked_where(s < upper, s)
slower = np.ma.masked_where(s > lower, s)
smiddle = np.ma.masked_where((s < lower) | (s > upper), s)
创建绘图
在这一步中，我们将使用上一步创建的掩码数组来创建绘图。我们将分别绘制每个掩码数组，并为每个数组使用不同的颜色。
fig, ax = plt.subplots()
ax.plot(t, smiddle, t, slower, t, supper)
plt.show()
总结
在本实验中，我们学习了如何在 Matplotlib 中使用掩码数组根据 y 值绘制不同颜色的线条。当可视化具有不同感兴趣区域的数据时，此技术有助于清晰区分不同区域，非常实用。