合并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)

# 使用groupby函数对列A进行分组，并应用agg函数
result = df.groupby('A').agg(lambda x: list(x))

print(result)

         B             C                D
A                                      
bar  [one, two, one]  [2, 4, 6]  [20, 40, 60]
foo  [one, two, two, one]  [1, 3, 5, 8]  [10, 30, 50, 80]