python - groupby 一列并计算另一个 pandas 中 5 以上的项目

Question

所以我有一个这样的 df:

NAME    TRY SCORE  
Bob   1st   3  
Sue   1st   7  
Tom   1st   3  
Max   1st   8  
Jay   1st   4  
Mel   1st   7  
Bob   2nd   4  
Sue   2nd   2  
Tom   2nd   6  
Max   2nd   4  
Jay   2nd   7  
Mel   2nd   8  
Bob   3rd   3  
Sue   3rd   5  
Tom   3rd   6  
Max   3rd   3  
Jay   3rd   4  
Mel   3rd   6 

我想统计每个人得分超过5分的次数？
进入一个新的 df2，看起来像这样:

NAME    COUNT  
Bob     0  
Sue     1  
Tom     2  
Mary    1  
Jay     1  
Mel     3  

我的尝试有很多 - 这是最新的

df2 = df.groupby('NAME')[['SCORE'] > 5].count().reset_index(name="count")

Answer 1

只需使用groupby 和sum

df.assign(SCORE=df.SCORE.gt(5)).groupby('NAME')['SCORE'].sum().astype(int).reset_index()
Out[524]: 
  NAME  SCORE
0  Bob      0
1  Jay      1
2  Max      1
3  Mel      3
4  Sue      1
5  Tom      2

或者我们使用 set_index 和 sum

df.set_index('NAME').SCORE.gt(5).sum(level=0).astype(int)