如何从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)

# 按列A进行分组，并对列C和列D求和
grouped = df.groupby('A').agg({'C': 'sum', 'D': 'sum'})

# 将生成的新列值添加到原始数据中
df = df.merge(grouped, left_on='A', right_index=True)

print(df)

     A    B  C   D
0  foo  one  16  170
2  foo  two  16  170
4  foo  two  16  170
6  foo  two  16  170
7  foo  one  16  170
1  bar  one   6   60
3  bar  two   6   60
5  bar  one   6   60