有没有办法在使用groupby.sum的同时保留其他列？

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.sum同时保留其他列
result = df.groupby(['A', 'B']).agg({'C': 'sum', 'D': 'mean'}).reset_index()

print(result)

     A    B   C     D
0  bar  one   2  40.0
1  bar  two   4  40.0
2  foo  one   9  50.0
3  foo  two  10  70.0