如何摆脱Pandas plot()的额外传奇条目?

时间:2013-03-14 13:47:59

标签: python matplotlib pandas

我有一个Pandas数据帧:

>>> drawdata
                   sales       mg
ID     RPT_Date                  
600809 20120331  22.1401  13.5591
       20120630  38.1565  22.8968
       20120930  52.5098  31.5909

以下命令绘制了一个条形图,如下所示img:

>>> drawdata.unstack('ID').plot(kind='bar', use_index=True)
<matplotlib.axes.AxesSubplot object at 0x035A9D70>

enter image description here

我的问题显示为注释:如何摆脱额外的传奇条目?

(此网址相关: How to barplot Pandas dataframe columns aligning by sub-index?

编辑:(有关HYRY方法的更多信息)


# -*- coding: utf-8 -*-

import itertools
import pandas as pd
import io

text = '''\
ID RPT_Date sales mg
000568 20120331 22.140 13.559
000568 20120630 38.156 22.896
000568 20120930 52.509 31.590
'''

drawdata = pd.read_csv(io.BytesIO(text), delimiter = ' ', converters = {0:str})
drawdata.set_index(['ID','RPT_Date'], inplace = True)

ax  = drawdata.unstack('ID').plot(kind='bar', use_index=True, legend=False)

display_acct_list = list(drawdata.columns)
legend_list = list(itertools.product(display_acct_list, list(set(drawdata.index.get_level_values('ID')))))
legend_list = [x[0] + '-' + x[1] for x in legend_list]

ax.legend(legend_list, loc='best', fancybox=True, shadow=True)

ax.set_xlabel("")

图表缺少图例中的“颜色框”。

enter image description here

1 个答案:

答案 0 :(得分:3)

您可以按以下方式关闭图例标题:

drawdata.unstack('ID').plot().get_legend().set_title("")

或者您可以自己绘制图例:

import pylab as pl
drawdata.unstack('ID').plot(legend=False)
pl.legend(loc="best")

关于你的第二个问题:

这是因为图中有一条虚线,如果你缩小一点,你会看到它。设置图例标签时,第一个标签将显示到虚线。您可以通过以下行修复它:

ax.legend(ax.containers, legend_list, loc='best', fancybox=True, shadow=True)