pandas:按多列分组。如何获得所有组合？

import pandas as pd
from itertools import combinations

# 创建一个示例DataFrame
df = pd.DataFrame({
    'A': ['a', 'a', 'b', 'b'],
    'B': ['x', 'y', 'x', 'y'],
    'C': [1, 2, 3, 4]
})

# 定义一个函数来获取所有组合
def get_combinations(group):
    combinations_list = list(combinations(group.values.tolist(), len(group.columns)))
    return pd.DataFrame(combinations_list, columns=group.columns)

# 按多列分组并获取所有组合
combinations_df = df.groupby(['A', 'B']).apply(get_combinations)

# 打印结果
print(combinations_df)