如何为pyplot.violinplot的均值正确设置图例标签？

问题场景 #

使用matplotlib.pyplot.violinplot绘制小提琴图时，希望显示均值并在图例中正确标注，但遇到两个问题：

1. 直接调用plt.legend()会重复标注分布，无法关联均值线条
2. 尝试通过violin['cmeans']设置图例时，报错'LineCollection' object is not iterable

用户示例代码：

import numpy as np
from matplotlib import pyplot as plt
plotData = [np.random.normal(loc = i,size = 100) for i in range(0,3)]
plotPos = range(0,3)
plt.violinplot(plotData,plotPos,showmeans=True,showextrema=False,widths=1,bw_method=0.5)
plt.legend(labels=['distribution','mean'])

错误尝试代码：

violin = plt.violinplot(plotData,plotPos,showmeans=True,showextrema=False,widths=1,bw_method=0.5)
plt.legend(handles=violin['cmeans'],labels=['mean'])

解决思路与代码 #

问题核心在于violin['cmeans']是LineCollection对象，而plt.legend()的handles参数需要接收可迭代的图例元素列表。我们可以手动创建对应分布和均值线的图例条目：

import numpy as np
from matplotlib import pyplot as plt
from matplotlib.lines import Line2D

# 生成数据
plotData = [np.random.normal(loc=i, size=100) for i in range(0, 3)]
plotPos = range(0, 3)

# 绘制小提琴图并获取返回对象
violin = plt.violinplot(plotData, plotPos, showmeans=True, showextrema=False, widths=1, bw_method=0.5)

# 创建自定义图例元素
# 取第一个小提琴的填充面作为分布的图例标识
dist_handle = violin['bodies'][0]
# 复刻均值线样式，生成Line2D对象作为均值的图例标识
mean_handle = Line2D(
    [], [], 
    color=violin['cmeans'].get_edgecolor()[0],
    linestyle=violin['cmeans'].get_linestyle()[0],
    linewidth=violin['cmeans'].get_linewidth()[0]
)

# 设置图例
plt.legend(handles=[dist_handle, mean_handle], labels=['分布', '均值'])

plt.show()

关键说明 #

• violin['bodies']是所有小提琴填充面的列表，取第一个即可代表整体分布样式
• LineCollection无法直接作为图例元素，通过Line2D手动复刻均值线的颜色、线型、线宽，就能生成符合要求的图例条目
• 最终图例会分别对应小提琴分布和均值线，解决重复标注与报错问题

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