使用y选项时,将图例添加到pandas.dataframe.plot

时间:2016-02-20 21:51:43

标签: python pandas matplotlib plot

如果我将DataFrame绘制为

df.plot()

然后每个行都出现在图例中,标识它应该对应的列。但是,如果我将其绘制为

df.plot(x=0, y=[2,3])

然后一切都正常绘制,但是没有图例标识哪一行是哪一列。

我试过了

df.plot(x=0, y=[2,3], label=['2','3'])
df.plot(x=0, y=[2,3], legend=['2','3'])

但没有任何作用。唯一的解决方法是设置

plt.legend(['2','3'])

之后,但我不确定图例列表的顺序与pandas用于绘图的顺序相同,所以我不知道图例是否真的与线条匹配。有没有办法让大熊猫绘制传说就是这种情况?

我正在使用pandas 0.14。

1 个答案:

答案 0 :(得分:1)

正如@unutbu所写,这适用于当前版本的Pandas。但是,在旧版本中,我发现只需将DataFrame“缩小”到相关列即可。

,例如,假设你有

df = pd.DataFrame({'a': [10, 2], 'b': [2, 3], 'c': [1, 3], 'd': [4, 5]})

然后将'b''c'绘制为'a'的函数:

df[['a', 'b', 'c']].plot(x='a');

这为您提供免费的图例订购。