python - 为 B 列的每个值添加 A 列的平均值

Question

假设您有以下 pandas 数据框 (df):

       A    B   C 
0      2    1   4
1      3    4   5
2      3    1   1
3      1    4   0 
4      5    0   1 

并且您想要为列 B 的每个不同值添加一个包含列 A 的平均值(或任何其他聚合函数)的列。您想要保留所有列。所以结果看起来像这样 (df_result):

       A    B   C   meanA_givenB
0      2    1   4        (2+3)/2
1      3    4   5        (3+1)/2
2      3    1   1        (2+3)/2
3      1    4   0        (3+1)/2
4      5    0   1          5  /1

我的做法如下:

import pandas as pd

meanA_givenB = df.groupby('B').A.mean()
df_result = pd.merge(df, pd.DataFrame(meanA_givenB), 
                     how='left', left_on='A', right_index=True,
                     suffixes=['','_meanA_givenB'])

必须有一种更清洁、更高效的方法来做到这一点。

Answer 1

您可以使用 transform :

>>> df['avg(A)|B'] = df.groupby('B')['A'].transform('mean')
>>> df
   A  B  C  avg(A)|B
0  2  1  4       2.5
1  3  4  5       2.0
2  3  1  1       2.5
3  1  4  0       2.0
4  5  0  1       5.0