python - 从另一个 GroupBy 对象创建一个 GroupBy 对象

Question

我按感兴趣的列对 df 进行分组:

grouped = df.groupby('columnA')

现在我只想保留至少有 5 名成员的组:

grouped.filter(lambda x: len(x) >= 5)

如果我尝试:

df2  = grouped.filter(lambda x: len(x) >= 5)

我得到的是 DataFrame，而不是 GroupBy 对象。那么如何在执行过滤后保留我的 GroupBy 对象呢？

Answer 1

一种解决方法是在过滤后的数据帧上调用groupby方法

grouped = grouped.filter(lambda x: len(x) >= 5).groupby('columnA')