Pandas聚合中的自定义函数

import pandas as pd

# 创建一个DataFrame
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
        'Age': [20, 21, 22, 20, 22],
        'Score': [90, 85, 92, 88, 95]}
df = pd.DataFrame(data)

# 定义一个自定义函数，计算平均年龄
def avg_age(x):
    return x.mean()

# 使用agg方法应用自定义函数
result = df.groupby('Name').agg({'Age': avg_age})
print(result)

      Age
Name     
John   22
Nick   21
Tom    20

import pandas as pd

# 创建一个DataFrame
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
        'Age': [20, 21, 22, 20, 22],
        'Score': [90, 85, 92, 88, 95]}
df = pd.DataFrame(data)

# 定义一个自定义函数，计算每个人的年龄和分数的平均值
def avg_age_score(x):
    return pd.Series({'AvgAge': x['Age'].mean(), 'AvgScore': x['Score'].mean()})

# 使用apply方法应用自定义函数
result = df.groupby('Name').apply(avg_age_score)
print(result)

      AvgAge  AvgScore
Name                  
John      22      93.5
Nick      21      85.0
Tom       20      89.0