Python pandas數據幀組基於條件

Question

我的問題很簡單，我有一個數據框，我根據列對結果進行groupby ，得到如下大小：

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)