在Python中,可以使用groupby
函数按照2个列参数进行分组。下面是一个使用lambda函数的示例代码:
import pandas as pd
# 创建示例数据
data = {'A': [1, 1, 2, 2, 3, 3],
'B': ['x', 'y', 'x', 'y', 'x', 'y'],
'C': [10, 20, 30, 40, 50, 60]}
df = pd.DataFrame(data)
# 使用lambda函数进行分组
groups = df.groupby(['A', 'B']).apply(lambda x: x['C'].sum())
print(groups)
输出结果:
A B
1 x 10
y 20
2 x 30
y 40
3 x 50
y 60
dtype: int64
在上面的代码中,groupby(['A', 'B'])
表示按照列'A'和列'B'进行分组。然后使用apply
函数对每个分组进行操作,lambda函数lambda x: x['C'].sum()
表示对每个分组中的'C'列进行求和操作。最后,将分组结果打印输出。
下一篇:按照2个列分组获取n条记录