在熊猫中的 groupby 和 nunique 之后过滤数据框

Question

我试过了df.groupby("item")["variable"].nunique()，它返回每个项目对象的唯一计数。

我想过滤以仅返回“变量”的计数 > 3 以 Groupby 项为条件...是否有方法？

Answer 1

当您希望groupby将 映射到输入的每一行时，请考虑transform：

df = df[df.groupby("item")["variable"].transform('nunique') > 3]