在 Matplotlib 中创建图例

简介

Matplotlib 是一个 Python 库，它能让用户创建各种类型的图表和绘图。本教程将指导你在 Matplotlib 绘图中创建图例的过程。

虚拟机使用提示

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

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

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

导入必要的库

在开始之前，我们需要导入必要的库。在这种情况下，我们将使用 Matplotlib 库。

import matplotlib.pyplot as plt

创建一个图形和子图

我们需要创建一个图形和子图来绘制数据。我们将创建一个包含两个子图的绘图。

fig = plt.figure()

ax = fig.add_subplot(211)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

ax = fig.add_subplot(223)
ax.plot([1, 2, 3], label="test1")
ax.plot([3, 2, 1], label="test2")

给绘图添加图例

现在我们要给绘图添加一个图例。在 Matplotlib 中有两种添加图例的方法。在这个例子中我们将两种方法都使用。

## 方法 1：将图例放置在子图上方
ax.legend(bbox_to_anchor=(0., 1.02, 1.,.102), loc='lower left',
           ncols=2, mode="expand", borderaxespad=0.)

## 方法 2：将图例放置在子图右侧
ax.legend(bbox_to_anchor=(1.05, 1), loc='upper left', borderaxespad=0.)

显示绘图

最后，我们将显示绘图。

plt.show()

总结

在本教程中，我们学习了如何在 Matplotlib 绘图中添加图例。我们使用了两种不同的方法来添加图例。第一种方法是将图例放置在子图上方，而第二种方法是将图例放置在子图右侧。