使用pandas groupby()添加公式的计算列

import pandas as pd

# 创建一个示例数据集
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
        'Subject': ['Math', 'English', 'Math', 'English', 'Math'],
        'Score': [80, 75, 90, 85, 95]}
df = pd.DataFrame(data)

# 使用groupby()函数按照Name列进行分组，并添加计算列AverageScore
df['AverageScore'] = df.groupby('Name')['Score'].apply(lambda x: x.mean())

print(df)

   Name  Subject  Score  AverageScore
0   Tom     Math     80          82.5
1  Nick  English     75          75.0
2  John     Math     90          90.0
3   Tom  English     85          82.5
4  John     Math     95          90.0