pandas对多个列进行分组，并选择新数据帧中group by使用的所有列

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)

# 对A和B列进行分组，并选择新数据帧中group by使用的所有列
grouped = df.groupby(['A', 'B']).agg(lambda x: x.tolist())

print(grouped)

           C          D
A   B                  
bar one    [bar]      [20]
    two    [bar]      [40]
foo one  [foo, foo]  [10, 80]
    two  [foo, foo]  [30, 70]