重组pandas数据帧，从多个层次创建新列

import pandas as pd

# 创建一个示例数据帧
data = {
    'A': [1, 2, 3, 4],
    'B': [5, 6, 7, 8],
    'C': [9, 10, 11, 12]
}
df = pd.DataFrame(data)

# 创建多级索引
df.columns = pd.MultiIndex.from_tuples([('Group1', 'A'), ('Group1', 'B'), ('Group2', 'C')])

# 使用assign()方法创建新列
df = df.assign(
    Group1_Sum=lambda x: x[('Group1', 'A')] + x[('Group1', 'B')],
    Group2_Sum=lambda x: x[('Group2', 'C')]
)

# 打印结果
print(df)

  Group1     Group2
       A  B       C
0      1  5       9
1      2  6      10
2      3  7      11
3      4  8      12

  Group1     Group2
       A  B       C
0      1  5       9
1      2  6      10
2      3  7      11
3      4  8      12