python - 在绘制 Pandas 之前过滤数据

Question

我想制作一个变量相对于另一个变量的条形图，并且仅显示高于阈值的条形图。

我的代码:

df.groupby('var1')['var2'].sum().where(lambda x : x > 100).plot.bar()

这会生成一个图，其中低于阈值的条消失，但仍显示在 x 轴中。

如何使过滤后的类别不再显示？

Answer 1

您可以过滤groupby的结果并绘图

df_sum = df.groupby('var1')['var2'].sum()
df_sum[df_sum > 100].plot.bar()