将图例添加到matplotlib箱线图中,在同一轴上有多个图

这个 boxplot 返回艺术家词典

结果:dict
将箱线图的每个组件映射到matplotlib列表的字典。线已创建Line2D实例。该字典具有以下键(假设垂直箱线图):

• boxes :箱线图的主体,显示四分位数和中值置信区间(如果启用)。
• [...]

使用 盒 ,您可以通过以下方式获得传奇艺术家:

ax.legend([bp1["boxes"][0], bp2["boxes"][0]], ['A', 'B'], loc='upper right')

完整示例:

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(1)

data1=np.random.randn(40,2)
data2=np.random.randn(30,2)

fig, ax = plt.subplots()
bp1 = ax.boxplot(data1, positions=[1,4], notch=True, widths=0.35, 
                 patch_artist=True, boxprops=dict(facecolor="C0"))
bp2 = ax.boxplot(data2, positions=[2,5], notch=True, widths=0.35, 
                 patch_artist=True, boxprops=dict(facecolor="C2"))

ax.legend([bp1["boxes"][0], bp2["boxes"][0]], ['A', 'B'], loc='upper right')

ax.set_xlim(0,6)
plt.show()

作为@ImportanceOfBeingErnest响应的补充,如果您在这样的for循环中绘图:

for data in datas:
    ax.boxplot(data, positions=[1,4], notch=True, widths=0.35, 
             patch_artist=True, boxprops=dict(facecolor="C0"))

不能将绘图保存为变量。在这种情况下,创建图例标签列表 legends ,将绘图追加到另一个列表中 elements 并使用列表理解为每一项添加图例:

labels = ['A', 'B']
colors = ['blue', 'red']
elements = []

for dIdx, data in enumerate(datas):
    elements.append(ax.boxplot(data, positions=[1,4], notch=True,\
    widths=0.35, patch_artist=True, boxprops=dict(facecolor=colors[dIdx])))

ax.legend([element["boxes"][0] for element in elements], 
    [labels[idx] for idx,_ in enumerate(datas)])