gca().lines与legend.get_lines()返回线条顺序是否一致？如何匹配？

顺序是否对应？ #

gca().lines和legend.get_lines()的顺序不一定一致：

• gca().lines按绘图先后顺序收集当前坐标轴上所有Line2D对象，包含所有绘制的线条（即使没有设置图例标签）。
• legend.get_lines()是图例中显示的线条副本，顺序由图例构建逻辑决定：
  • 自动生成图例时，顺序是检测到的带有效标签（非_nolegend_）的线条顺序；
  • 手动指定handles参数创建图例时，顺序完全由传入的handles列表决定，和绘图顺序无关。

如何从gca().lines获取对应图例线条？ #

有两种可靠的关联方式：

1. 通过标签（label）匹配 #

利用线条的label属性关联，适合自动生成图例或标签唯一的场景：

import matplotlib.pyplot as plt

# 假设已完成绘图并创建图例
ax = plt.gca()
legend = ax.get_legend()

# 构建绘图线条到图例线条的映射
line_legend_map = {}
for plot_line in ax.lines:
    line_label = plot_line.get_label()
    # 跳过无图例的线条（默认无label的线条会被标记为_nolegend_）
    if line_label == '_nolegend_':
        continue
    # 匹配图例中同标签的线条
    for legend_line in legend.get_lines():
        if legend_line.get_label() == line_label:
            line_legend_map[plot_line] = legend_line
            break

2. 通过图例的原始句柄（handles）关联 #

legend.get_handles()返回的是绘图时的原始线条对象（而非图例中的副本），是更直接的关联方式：

import matplotlib.pyplot as plt

ax = plt.gca()
legend = ax.get_legend()
legend_handles = legend.get_handles()

# 遍历绘图线条，找到其在图例中的对应线条
for plot_line in ax.lines:
    if plot_line in legend_handles:
        # 获取该线条在图例中的索引
        handle_idx = legend_handles.index(plot_line)
        # 通过索引获取对应的图例线条
        corresponding_legend_line = legend.get_lines()[handle_idx]
        # 后续可对对应线条操作

注意：如果手动指定了图例的handles参数，legend_handles的顺序就是手动传入的顺序，此时只能通过对象本身或标签来匹配关联。

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