Python绘图-遍历要素和绘图的唯一值

时间:2018-08-19 16:25:29

标签: python pandas matplotlib

我的熊猫数据框中有一个功能,可以(可能)具有多个级别。我可以使用以下方法获得唯一的级别:

reasons = df["Reason"].unique()

我可以执行以下操作在一张图表上绘制所有级别的计数:

for reason in reasons: df[df['Reason']== reason].groupby('date').count()['twp'].plot() plt.title(reason) plt.tight_layout()

如何修改此代码,以便为每个级别创建单独的图形?此外,我还要确保可伸缩性。 (例如,如果当前唯一级别的数量为3,但是将来我们可以最终得到5个级别)。

提前谢谢!

1 个答案:

答案 0 :(得分:1)

如Bazingaa所述,请使用子图:

reasons = df["Reason"].unique()
fig, axes = plt.subplots(len(reasons), 1) 

for reason, ax in zip(reasons, axes):
    df[df['Reason']== reason].groupby('date').count()['twp'].plot(ax=ax)
    ax.set_title(reason)

plt.tight_layout()