如何为 Matplotlib 图例添加标题？

Matplotlib 是一个用于绘制数据可视化的Python库。它提供了许多强大的绘图功能，其中之一是图例（Legend）。图例通常用于标识不同的数据系列，以帮助读者理解图表中的含义。有时候，为了更好地解释图例的含义，我们可能需要为图例添加一个标题。

本文将介绍如何使用 Matplotlib 为图例添加标题。

方法一：使用 legend 函数中的 title 参数

Matplotlib 的 legend 函数接受一个可选的 title 参数，该参数用于设置图例的标题。我们可以通过将标题文本传递给 title 参数，实现为图例添加标题的功能。

import matplotlib.pyplot as plt

# 创建示例数据
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

# 绘制曲线
plt.plot(x, y1, label='y = x^2')
plt.plot(x, y2, label='y = x')

# 添加图例标题
plt.legend(title='函数关系')

# 显示图表
plt.show()

运行上述代码将绘制带有标题的图例。

方法二：使用 set_title 方法

另一种方式是使用图例对象的 set_title 方法来设置标题。我们需要先获取图例对象，然后调用其 set_title 方法，传入标题文本。

import matplotlib.pyplot as plt

# 创建示例数据
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

# 绘制曲线
plt.plot(x, y1, label='y = x^2')
plt.plot(x, y2, label='y = x')

# 添加图例
legend = plt.legend()

# 设置图例标题
legend.set_title('函数关系')

# 显示图表
plt.show()

运行上述代码将绘制带有标题的图例。

无论使用哪种方法，都可以在 Matplotlib 图例中添加一个标题来更好地解释图表中的数据系列。希望本文对你理解如何为 Matplotlib 图例添加标题有所帮助！