python - 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

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

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

它是一个转变我的男人

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