如何从多个条形图中的图例中删除一个条形标签

时间:2018-11-08 08:57:24

标签: python pandas matplotlib

我在条形图中有两个酒吧。但是,我只想显示图例中的标签之一。示例:

In [190]: import pandas as pd

In [191]: import matplotlib.pyplot as plt

In [192]: import seaborn as sns

In [193]: from matplotlib.colors import ListedColormap

In [194]: my_cmap = ListedColormap(sns.color_palette("GnBu_d", 4).as_hex())

In [195]: df1 = pd.DataFrame([[200, 0.0, 0.0, 0.0], [0.0, 120, 60, 20]], columns=['Total','A', 'B', 'C'])

In [196]: ax = df1.plot(colormap=my_cmap, title='Allocation per bucket', kind='bar', stacked=True, alpha=0.8)

In [197]: ax.title.set_size(30)

In [198]: ax.set_ylabel('Value', fontsize=15)
Out[198]: Text(0,0.5,'Value')

In [199]: ax.set_xticklabels(['Total', 'Split'])
Out[199]: [Text(0,0,'Total'), Text(0,0,'Split')]

In [200]: box = ax.get_position()

In [201]: ax.set_position([box.x0, box.y0, box.width * 0.8, box.height])

In [202]: ax.legend(loc='center left', bbox_to_anchor=(1, 0.5), fontsize=13, prop={'size': 13})
Out[202]: <matplotlib.legend.Legend at 0x7f1ce0b4e050>

In [203]: plt.xticks(rotation=0)
Out[203]: (array([0, 1]), <a list of 2 Text xticklabel objects>)

In [204]: plt.show()

如您所见,这导致了下图

enter image description here

我希望将Total从图例中删除,仅在其中显示ABC。我该如何实现?

2 个答案:

答案 0 :(得分:0)

只需在['A','B','C']中添加ax.legend()。您的代码行应如下所示:

ax.legend(['A','B','C'], loc='center left', bbox_to_anchor=(1, 0.5), fontsize=13, prop={'size': 13})

您将从图例中删除总计

答案 1 :(得分:0)

另一个答案只是将前三个图例标签重命名为['A','B','C'],而没有实际删除“总计”(并且不显示最后一个图例条目)。通过查看图例中的颜色可以看出这一点。 “总计”的颜色仍然是图例中的第一个条目,但是已将其重命名为“ A”。我认为这不是理想的结果。

您可以使用ax.get_legend_handles_labels()获取图例句柄和标签的列表,然后将结果切片为不包括第一项:

handles, labels = ax.get_legend_handles_labels()
ax.legend(handles=handles[1:], labels=labels[1:],
          loc='center left', bbox_to_anchor=(1, 0.5), fontsize=13, prop={'size': 13})

哪个给:

enter image description here