python - 使用 secondary_y 轴绘制 groupby 数据

Question

我想绘制 12 个图表(每月一个图表)，包括左侧 y 轴上的列 'A' 和 'B' 以及列 ' C'在右边。

下面的代码将所有内容绘制在左侧。

import pandas as pd
index=pd.date_range('2011-1-1 00:00:00', '2011-12-31 23:50:00', freq='1h')
df=pd.DataFrame(np.random.rand(len(index),3),columns=['A','B','C'],index=index)

df2 = df.groupby(lambda x: x.month)
for key, group in df2:
    group.plot()

如何分隔列并使用类似这样的东西:group.plot({'A','B':style='g'},{'C':secondary_y=True}) ？

Answer 1

您可以捕获 Pandas plot() 命令返回的轴，并再次使用它在右轴上专门绘制 C。

index=pd.date_range('2011-1-1 00:00:00', '2011-12-31 23:50:00', freq='1h')
df=pd.DataFrame(np.random.randn(len(index),3).cumsum(axis=0),columns=['A','B','C'],index=index)

df2 = df.groupby(lambda x: x.month)
for key, group in df2:
    ax = group[['A', 'B']].plot()
    group[['C']].plot(secondary_y=True, ax=ax)

要获取单个图例中的所有行，请参阅: Legend only shows one label when plotting with pandas