我的熊猫数据框中有一个功能,可以(可能)具有多个级别。我可以使用以下方法获得唯一的级别:
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个级别)。
提前谢谢!
答案 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()