我有一个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>
我的问题显示为注释:如何摆脱额外的传奇条目?
(此网址相关: 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("")
图表缺少图例中的“颜色框”。
答案 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)