python - 分组依据，在 Pandas 中

Question

select df.id, count(distinct airports) as num
from df
group by df.id
having count(distinct airports) > 3

我正在尝试在 Python pandas 中执行与上述相同的操作。我尝试了 filter、nunique、agg 的不同组合，但没有任何效果。有什么建议吗？

例如: df

df   
id     airport
1      lax
1      ohare
2      phl
3      lax
2      mdw
2      lax
2      sfw
2      tpe

所以我希望结果是:

id     num
2      5

Answer 1

您可以使用 SeriesGroupBy.nunique与 boolean indexing或 query :

s = df.groupby('id')['airport'].nunique()
print (s)
id
1    2
2    5
3    1
Name: airport, dtype: int64

df1 = s[s > 3].reset_index()
print (df1)
   id  airport
0   2        5

或者:

df1 = df.groupby('id')['airport'].nunique().reset_index().query('airport > 3')
print (df1)
   id  airport
1   2        5