pandas 按函数分组应用于列

python
import pandas as pd
# 创建示例数据
data = {'姓名': ['张三', '李四', '王五', '赵六'],
        '年龄': [25, 30, 35, 40]}
df = pd.DataFrame(data)
print(df)

  姓名  年龄
0  张三  25
1  李四  30
2  王五  35
3  赵六  40

python
# 计算年龄的平均值
average_age = df['年龄'].apply(lambda x: x.mean())
print(average_age)

0    32.5
Name: 年龄, dtype: float64

python
# 将计算结果添加为新的列
df = df.assign(平均年龄=average_age)
print(df)

  姓名  年龄  平均年龄
0  张三  25  32.5
1  李四  30  32.5
2  王五  35  32.5
3  赵六  40  32.5