使用 Matplotlib 为图表添加注释

简介

本教程将指导你完成使用 Matplotlib 为图表添加注释的过程。为图表添加注释是突出显示图表上特定特征或数据点的有用方法。在本教程中，我们将演示如何使用指向给定坐标的箭头为图表添加注释。

虚拟机提示

虚拟机启动完成后，点击左上角切换到“笔记本”标签，以访问 Jupyter Notebook 进行练习。

有时，你可能需要等待几秒钟让 Jupyter Notebook 完成加载。由于 Jupyter Notebook 的限制，操作验证无法自动化。

如果你在学习过程中遇到问题，随时向 Labby 提问。课程结束后提供反馈，我们将立即为你解决问题。

导入库

在开始之前，我们需要导入必要的库。在本教程中，我们将使用 Matplotlib 和 Numpy。

import matplotlib.pyplot as plt
import numpy as np

创建一个图表

接下来，我们将使用 Matplotlib 创建一个图表。在这个例子中，我们将在一系列值上绘制余弦函数。

fig, ax = plt.subplots()

t = np.arange(0.0, 5.0, 0.01)
s = np.cos(2*np.pi*t)
line, = ax.plot(t, s, lw=2)

为图表添加注释

现在，我们将通过添加一个指向特定坐标的箭头来为图表添加注释。在这个例子中，我们将添加一个指向余弦函数局部最大值的箭头。

ax.annotate('local max', xy=(2, 1), xytext=(3, 1.5),
            arrowprops=dict(facecolor='black', shrink=0.05),
            )

ax.annotate() 函数接受几个参数。第一个参数是将显示在图表上的文本。xy 参数指定我们要注释的点的坐标。xytext 参数指定文本的坐标。arrowprops 参数是一个字典，用于指定箭头的属性。

设置图表界限

最后，我们将设置图表的界限，以确保注释的点是可见的。

ax.set_ylim(-2, 2)
plt.show()

总结

在本教程中，你已经学会了如何使用 Matplotlib 为图表添加注释。我们首先导入了必要的库并创建了一个图表。然后，我们通过添加一个指向特定坐标的箭头为图表添加了注释。最后，我们设置了图表的界限，以确保注释的点是可见的。