Python: Removing Rows on Count condition

前端 未结 4 1130
攒了一身酷
攒了一身酷 2020-12-09 10:07

I have a problem filtering a pandas dataframe.

city 
NYC 
NYC 
NYC 
NYC 
SYD 
SYD 
SEL 
SEL
...

df.city.value_counts()

I woul

4条回答
  •  被撕碎了的回忆
    2020-12-09 10:20

    Here you go with filter

    df.groupby('city').filter(lambda x : len(x)>3)
    Out[1743]: 
      city
    0  NYC
    1  NYC
    2  NYC
    3  NYC
    

    Solution two transform

    sub_df = df[df.groupby('city').city.transform('count')>3].copy() 
    # add copy for future warning when you need to modify the sub df
    

提交回复
热议问题