Python pandas数据帧组基于条件

Question

我的问题很简单，我有一个数据框，我根据列对结果进行分组，得到如下大小：

df.groupby('column').size()

现在的问题是我只想要大小大于X的那些。我想知道我是否可以使用lambda函数或类似的东西来做它？我已经尝试过了：

df.groupby('column').size() > X

它打印出一些True和False值。

由于

Answer 1

分组结果是常规DataFrame，因此只需像往常一样过滤结果：

 import pandas as pd

 df = pd.DataFrame({'a': ['a', 'b', 'a', 'a', 'b', 'c', 'd']})
 after = df.groupby('a').size()
 >> after
 a
 a    3
 b    2
 c    1
 d    1
 dtype: int64

 >> after[after > 2]
 a
 a    3
 dtype: int64

Answer 2

试试这段代码：

df.groupby('column').filter(lambda group: group.size > X)