我在条形图中有两个酒吧。但是,我只想显示图例中的标签之一。示例:
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()
如您所见,这导致了下图
我希望将Total
从图例中删除,仅在其中显示A
,B
和C
。我该如何实现?
答案 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})
哪个给: