如何将Pandas groupby结果广播到所有行？

import pandas as pd

# 创建示例DataFrame
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
        'B': ['one', 'one', 'two', 'two', 'two', 'one', 'two', 'one'],
        'C': [1, 2, 3, 4, 5, 6, 7, 8]}
df = pd.DataFrame(data)

# 使用groupby函数对'A'列进行分组，并计算每个分组的平均值
grouped = df.groupby('A')['C'].transform('mean')

# 将分组结果广播到所有行
df['group_mean'] = grouped

print(df)

     A    B  C  group_mean
0  foo  one  1    4.666667
1  bar  one  2    4.000000
2  foo  two  3    4.666667
3  bar  two  4    4.000000
4  foo  two  5    4.666667
5  bar  one  6    4.000000
6  foo  two  7    4.666667
7  foo  one  8    4.666667