为pandas groupby中的不同要素分配不同的聚合函数

import pandas as pd

# 创建一个示例数据集
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
        'B': ['one', 'one', 'two', 'two', 'two', 'one', 'two', 'one'],
        'C': [1, 2, 3, 4, 5, 6, 7, 8],
        'D': [10, 20, 30, 40, 50, 60, 70, 80]}
df = pd.DataFrame(data)

# 创建一个字典，指定要素和对应的聚合函数
agg_dict = {'C': 'sum', 'D': 'mean'}

# 使用groupby和agg函数进行分组和聚合
result = df.groupby('A').agg(agg_dict)

print(result)

      C     D
A            
bar   6  40.0
foo  16  40.0