pandas分组聚合代码

以下是pandas分组聚合代码的示例：

假设我们有一个数据集，其中包含客户的购买记录，包括客户ID、购买日期、购买金额和购买类别。我们想要按客户ID对数据进行分组，并计算每个客户的总购买金额和平均购买金额。

import pandas as pd

# 创建数据集
data = {'customer_id': ['C1', 'C1', 'C2', 'C2', 'C2', 'C3', 'C3'],
        'date': ['2021-01-01', '2021-02-01', '2021-01-01', '2021-02-01', '2021-03-01', '2021-01-01', '2021-02-01'],
        'amount': [100, 200, 150, 300, 250, 100, 150],
        'category': ['A', 'B', 'A', 'B', 'A', 'B', 'A']}

df = pd.DataFrame(data)

# 按客户ID分组并计算总购买金额和平均购买金额
result = df.groupby('customer_id').agg({'amount': ['sum', 'mean']})

print(result)

输出结果为：

            amount           
               sum        mean
customer_id                   
C1             300  150.000000
C2             700  233.333333
C3             250  125.000000