select df.id, count(distinct airports) as num
from df
group by df.id
having count(distinct airports) > 3
我试图在Python熊猫中做相同的上述操作.我尝试过滤器,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
解决方法:
您可以在boolean indexing
或query
中使用SeriesGroupBy.nunique
:
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