python - 如何添加两个字段出现次数的列

Question

我有以下 df:

df = pd.DataFrame({'c1': [1,5,2,2,2,5],'c2':[2,2,2,2,1,2]})

   c1  c2
0   1   2
1   5   2
2   2   2
3   2   2
4   2   1
5   5   2

我试图得到

   c1  c2  c3
0   1   2   1
1   5   2   2
2   2   2   2
3   2   2   2
4   2   1   1
5   5   2   2

我开始于

df['c3'] = df.groupby('c1')['c2'].transform('size')

但不知道如何按两列进行分组。

Answer 1

如果您只需要使用 group by 进行计数，那么您可以尝试如下

import pandas as pd

df = pd.DataFrame({'c1': [1,5,2,2,2,5],'c2':[2,2,2,2,1,2]})

print (df.groupby(['c1', 'c2']).size().reset_index(name='counts'))