Python Matplotlib：在XY绘图后于函数外为多阶段数据线条设标签

嘿，我来帮你搞定这个Matplotlib的标签设置问题～首先得提一句，你代码里有两个小细节需要修正：一是ax.plot的第二个参数写错啦，应该用list_of_y_points[i]而不是list_of_x_points[i]；二是linecolor这个参数名不对，Matplotlib里正确的参数是color哦。

回到你的核心需求——在绘图函数外部给线条设置标签，这里有两种实用的实现方式，你可以根据场景选择：

方法1：提前准备标签，绘图时直接绑定（最常用） #

如果你已经知道每条折线对应的标签内容，直接先定义一个标签列表，循环绘图的时候把label参数加上就行，最后调用图例就搞定了：

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(11,8.5))
ax = fig.add_subplot(111)
list_of_x_points = [[1,2,3],[4,5,6,7,8],[9,10,11,12]]
list_of_y_points = [[1,1,1],[2,2,2,2,2],[3,3,3,3]]
# 提前定义好每条线对应的标签
line_labels = ["测试阶段1", "测试阶段2", "测试阶段3"]

for i in range(len(list_of_x_points)):
    # 绘图时直接把标签传进去
    ax.plot(list_of_x_points[i], list_of_y_points[i], color="b", linewidth=1, linestyle='dashed', label=line_labels[i])

# 显示图例，默认位置在右上角
ax.legend()
plt.show()

方法2：绘图完成后，批量给线条加标签（纯外部设置） #

如果确实需要在所有线条都绘制完成后再设置标签，也完全没问题——Matplotlib会把所有绘制的线条存在ax.lines里，你直接遍历这个列表，给每个线条对象设置label属性就行：

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(11,8.5))
ax = fig.add_subplot(111)
list_of_x_points = [[1,2,3],[4,5,6,7,8],[9,10,11,12]]
list_of_y_points = [[1,1,1],[2,2,2,2,2],[3,3,3,3]]

# 先把所有线条画出来，不用管标签
for i in range(len(list_of_x_points)):
    ax.plot(list_of_x_points[i], list_of_y_points[i], color="b", linewidth=1, linestyle='dashed')

# 现在来外部设置标签
line_labels = ["测试阶段1", "测试阶段2", "测试阶段3"]
# 用zip把线条和标签配对，逐个设置
for line, label in zip(ax.lines, line_labels):
    line.set_label(label)

# 别忘了调用legend显示标签，这里可以指定图例位置
ax.legend(loc="lower right")
plt.show()

小提醒 #

• 一定要保证标签列表的长度和你画的线条数量一致，不然会出现索引错误哦
• 如果需要区分不同线条，也可以在设置标签的时候顺便改改线条颜色，比如line.set_color("r")，这样看图例的时候更清晰
• ax.legend()还有很多参数可以调，比如fontsize改字体大小，frameon=False去掉图例边框，都可以按需调整

内容的提问来源于stack exchange，提问作者sgatta