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


问题内容

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

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


问题答案:

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

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