咨询Pandas与Matplotlib绘图时图例标签顺序异常的原因

问题现象 #

在用Pandas绘制DataFrame柱状图后添加水平参考线，按['case0', 'case1', 'case2'] + ['line label']的顺序设置图例标签时，出现标签与图形不匹配的异常；反而用['line label'] + ['case0', 'case1', 'case2']的顺序设置，才能得到正确的对应关系。

相关代码 #

import numpy as np, pandas as pd
import matplotlib.pyplot as plt

data = np.random.rand(5,3)
figsize = (8, 4)
pal = ['#1f77b4', '#ff7f0e', '#2ca02c']

plt.figure(figsize=figsize, dpi=120)
ax = pd.DataFrame(data).plot.bar(color=pal)
ax.axhline(y=0.4, lw=1, ls='--', c='k', label='line')
# 异常的设置方式
ax.legend(['case0', 'case1', 'case2'] + ['line label'], fontsize=6, loc='upper left')

原因解析 #

核心问题出在Matplotlib收集图例元素的优先级规则，而非你执行绘图的顺序：

• Pandas绘制的柱状图属于「Patch类」图形对象（比如矩形柱子），而axhline生成的是「Line2D类」线条对象。
• Matplotlib默认会按类型优先级收集图例元素：先收集所有Line2D对象，再收集Patch对象，完全不考虑它们被添加到坐标轴的先后顺序。
• 在你的代码里，虽然先画了柱状图，但Matplotlib收集图例元素时，会把后添加的水平参考线（Line2D）放在最前面，再依次排列三个柱状图系列（Patch）。内部的元素顺序实际是[水平线, case0柱子, case1柱子, case2柱子]。
• 当你按['case0', 'case1', 'case2', 'line label']传标签时，第一个标签会贴到水平线上，最后一个标签贴到case2柱子上，自然出现错位；反过来按['line label', 'case0', 'case1', 'case2']传标签，才能和内部元素顺序一一对应，得到正确的图例。

验证方法 #

你可以在调用ax.legend()之前，打印Matplotlib收集的元素和默认标签，直观看到顺序：

handles, labels = ax.get_legend_handles_labels()
print("元素顺序:", [h.get_label() for h in handles])
# 输出会是 ['line', '0', '1', '2']，对应水平线、三个默认列名的柱子

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