计数按列分组的每个行的最佳方法

时间:2019-11-28 02:35:45

标签: python pandas

按两列分组后df.groupby([“ id”,“ b”]) 现在我要查找“ id”,其中有5行以上。

因此在下面的df中,

id = 4有2行
id = 4有3行。

                count   
id          b           
4         1568    1 
          4167    1 
5         1100    1 
          1832    2 
          1969    5 

我只是reset_index()

并在每行上获取id值,然后将其添加。

2 个答案:

答案 0 :(得分:1)

尝试

rows = df.groupby("id")["b"].apply(lambda x: len(list(x)))

输出

id
4    2
5    3

答案 1 :(得分:0)

IIUC

df.groupby(level=0).size()