Pandas -如何进行分组，其中新列的结果是(列的总和)/(分组的iten数)？

import pandas as pd

# 创建DataFrame对象
df = pd.DataFrame({
    'group_col': ['A', 'A', 'B', 'B', 'B'],
    'sum_col': [1, 2, 3, 4, 5]
})

# 分组并计算新列
grouped = df.groupby('group_col')
result = grouped['sum_col'].transform(lambda x: x.sum() / len(x))

# 将结果添加到原DataFrame中
df['new_col'] = result

# 打印结果
print(df)

  group_col  sum_col  new_col
0         A        1      1.5
1         A        2      1.5
2         B        3      4.0
3         B        4      4.0
4         B        5      4.0