print(data.groupby(by=['ID'], as_index=False).count())
只能产生分组频数统计
----------------------
0 a
1 a
2 b
但是我期望结果是
0 a 2
1 a 2
2 b 1
这种如何解决
1个回答
用transform,然后可以把结果作为新的一列count_ID插入到dataframe里
data['count_ID'] = data.groupby('ID')['ID'].transform('count')