Pandas goupby和aggregate as list不起作用

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和aggregate函数进行分组和聚合操作
result = df.groupby(['A', 'B']).aggregate({'C': 'sum', 'D': 'mean'})

# 自定义函数，将聚合结果转化为列表形式
def to_list(series):
    return list(series)

# 使用apply函数，将聚合结果转化为列表形式
result['C'] = result['C'].apply(to_list)
result['D'] = result['D'].apply(to_list)

print(result)

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