Python 绘图 - 循环遍历特征和绘图的唯一值

Question

我的 pandas 数据框中有一个功能，可以(可能)具有多个级别。我可以使用以下方法获得独特的级别:

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

我可以通过执行以下操作在单个图表上绘制所有级别的计数:

原因中的原因:
df[df['Reason']== Reason].groupby('date').count()['twp'].plot()
plt.标题(原因)
plt.tight_layout()

如何修改此代码，以便它可以为每个级别创建单独的图表？此外，我还想确保可扩展性。 (例如，如果当前唯一级别的数量为 3，但将来我们可能会达到 5 个级别)。

提前谢谢您!

Answer 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()