使用 Matplotlib 为图表添加注释

简介

在本实验中，你将学习如何使用Matplotlib为图表添加注释。注释是指向图表添加文本、箭头和形状以提供更多背景信息或突出显示特定感兴趣点的过程。

虚拟机提示

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

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

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

导入Matplotlib

在我们开始使用Matplotlib为图表添加注释之前，必须先导入该库。在这一步中，我们将导入Matplotlib并创建一个简单的图表用于注释。

import matplotlib.pyplot as plt

## 创建一个简单的图表
fig, ax = plt.subplots()
ax.plot([0, 1, 2, 3, 4], [0, 1, 4, 9, 16])
plt.show()

添加文本注释

注释的最简单形式是向图表添加文本。在这一步中，我们将向在上一步中创建的图表添加文本。

## 添加文本注释
ax.text(2, 10, "Important Point", fontsize=12, color='red')
plt.show()

添加箭头注释

箭头可用于指出图表中的特定特征或趋势。在这一步中，我们将向图表添加一个指向最大值的箭头。

## 找到最大值
y = [0, 1, 4, 9, 16]
max_index = y.index(max(y))
xmax = max_index
ymax = y[max_index]

## 添加箭头注释
ax.annotate('Maximum Value', xy=(xmax, ymax), xytext=(xmax, ymax + 5),
            arrowprops=dict(facecolor='black', shrink=0.05))
plt.show()

添加形状注释

形状可用于吸引对图表特定区域的关注。在这一步中，我们将添加一个矩形来突出显示(x = 1)和(x = 3)之间的区域。

## 添加形状注释
ax.axvspan(1, 3, facecolor='gray', alpha=0.2)
plt.show()

总结

在这个实验中，你学习了如何使用Matplotlib为图表添加注释。你学会了如何添加文本、箭头和形状，以提供更多背景信息或突出显示特定的感兴趣点。有了这些工具，你可以创建更具信息性和视觉吸引力的图表与他人分享。