在绘制熊猫之前过滤数据

Question

我想绘制一个变量相对于另一个变量的条形图，仅显示阈值以上的条形图。

我的代码：

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

这将生成一个曲线图，其中阈值以下的条消失，但仍在x轴上显示。

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

Answer 1

您可以过滤分组结果和绘图的结果

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