Pandas：当列相等时获取不同行的平均值

Question

我试图找出不同行中值的平均值，按其他列中的相似性分组。示例：

In [14]: pd.DataFrame({'col1':[1,2,1,2], 'col2':['A','C','A','B'], 'col3':[1, 5, 6, 9]})
Out[14]: 
   col1 col2  col3
0     1    A     1
1     2    C     5
2     1    A     6
3     2    B     9

我想要的是用 col3 的方式为 col1 和 col2 的组合匹配的所有行添加一列。期望的输出：

Out[14]: 
   col1 col2  col3   mean
0     1    A     1    3.5
1     2    C     5    5
2     1    A     6    3.5
3     2    B     9    9

我用groupby 和apply 尝试了几件事，但无法得到正确的结果。

Answer 1

这是transform我的男人

df['mean'] = df.groupby(['col1','col2']).col3.transform('mean')