Matplotlib PyPlot 折线图图例绘制指南：添加、自定义、常见问题解答

在 Matplotlib 中为 PyPlot 折线图添加图例

在数据可视化中，为图形添加图例对于解释不同数据序列之间的关系至关重要。Matplotlib 的 PyPlot 为你提供了方便的方式来为折线图添加图例。

创建折线图

首先，你需要创建一个包含要可视化数据的折线图。使用 pyplot.plot() 函数指定数据点和线条颜色。

import matplotlib.pyplot as plt

# 创建数据
length = [10, 20, 30, 40, 50]
bubble = [0.1, 0.2, 0.3, 0.4, 0.5]
ins = [0.01, 0.02, 0.03, 0.04, 0.05]
merge_r = [0.001, 0.002, 0.003, 0.004, 0.005]
merge_i = [0.0001, 0.0002, 0.0003, 0.0004, 0.0005]

# 创建折线图
plt.plot(length, bubble, 'b-')  # 蓝色虚线
plt.plot(length, ins, 'r-')  # 红色虚线
plt.plot(length, merge_r, 'g+')  # 绿色加号
plt.plot(length, merge_i, 'p-')  # 紫色虚线

添加图例

要为折线图添加图例，可以使用 pyplot.legend() 函数。此函数将自动从线条标签生成图例条目。

plt.legend()

自定义图例

你可以通过指定标签参数来自定义图例条目。这对于提供更具体的信息很有用。

plt.plot(length, bubble, 'b-', label="Bubble")
plt.plot(length, ins, 'r-', label="Insertion")
plt.plot(length, merge_r, 'g+', label="Merge Recursive")
plt.plot(length, merge_i, 'p-', label="Merge Iterative")

plt.legend()

结论

为 PyPlot 折线图添加图例是一种简单的方法，可以为你的可视化提供上下文和清晰度。通过自定义图例条目，你可以向你的观众提供有关所显示数据的重要信息。

常见问题解答

1. 如何改变图例的位置？
   你可以在 pyplot.legend() 函数中使用 loc 参数指定图例的位置。

2. 如何更改图例条目的字体大小？
   使用 fontsize 参数指定图例条目的字体大小。

3. 如何更改图例条目的颜色？
   使用 facecolor 和 edgecolor 参数更改图例条目的颜色。

4. 如何隐藏图例？
   设置 pyplot.legend() 函数的 visible 参数为 False。

5. 如何将图例保存为图像？
   使用 pyplot.savefig() 函数将图例保存为图像。