Pandas按多列分组并保留所有其他列

import pandas as pd

# 创建一个示例DataFrame
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'进行分组，并保留所有其他列
grouped = df.groupby(['A', 'B'], as_index=False)

# 对分组后的数据进行聚合操作，例如计算每组的平均值
result = grouped.mean()

# 打印结果
print(result)

     A    B    C     D
0  bar  one  40  40.0
1  bar  two  40  50.0
2  foo  one  10  45.0
3  foo  two  5   70.0