Pandas:如何在lambda公式中使用(df.groupby)

import pandas as pd

# 创建DataFrame对象
df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
                   'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
                   'C': [1, 2, 3, 4, 5, 6, 7, 8],
                   'D': [10, 20, 30, 40, 50, 60, 70, 80]})

# 按照列'A'进行分组
grouped = df.groupby('A')

# 按照多个列进行分组
grouped = df.groupby(['A', 'B'])

# 对分组后的数据求和
sum_result = grouped.sum()

# 对分组后的数据计数
count_result = grouped.count()

# 对分组后的数据求平均值
mean_result = grouped.mean()

# 自定义lambda公式
custom_result = grouped.agg(lambda x: x.max() - x.min())