Pandas: remove group from the data when a value in the group meets a required condition

前端 未结 3 1813
盖世英雄少女心
盖世英雄少女心 2021-01-04 03:44

I have groupings of values in the data and within each group, I would like to check if a value within the group is below 8. If this condition is met, the entire

3条回答
  •  Happy的楠姐
    2021-01-04 04:07

    Based on what you described in the question, as long as there is at least one value is below 8 within the group, then that group should be dropped. So the equivalent statement is that as long as the minimum value within that group is below 8, that group should be dropped.

    By using the filter feature, the actual code can be reduced to only one line, please refer to Filtration, you may use the following code:

    dfnew = df.groupby('Groups').filter(lambda x: x['Count'].min()>8 )
    dfnew.reset_index(drop=True, inplace=True) # reset index
    dfnew = dfnew[['Groups','Count']] # rearrange the column sequence
    print(dfnew)
    
    Output:
       Groups  Count
    0       2     12
    1       2     15
    2       2     21
    

提交回复
热议问题