如何在groupby上应用自己构建函数

import pandas as pd

# 创建一个示例DataFrame
data = {'Category': ['A', 'A', 'B', 'B', 'A'],
        'Value': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)

# 自定义函数，计算每个分组的平均值
def custom_function(group):
    return group.mean()

# 在Category列上进行groupby，并应用自定义函数
result = df.groupby('Category')['Value'].apply(custom_function)

print(result)

Category
A    2.666667
B    3.500000
Name: Value, dtype: float64