获取pandas分组的DataFrame的列和组的值

import pandas as pd

# 创建一个示例DataFrame
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar'],
        'B': ['one', 'one', 'two', 'two', 'two', 'one'],
        'C': [1, 2, 3, 4, 5, 6],
        'D': [10, 20, 30, 40, 50, 60]}
df = pd.DataFrame(data)

# 按照列'A'进行分组
grouped = df.groupby('A')

# 定义一个函数，用于获取分组的列和组的值
def get_group_info(group):
    # 获取分组的列
    columns = group.columns.tolist()
    
    # 获取组的值
    group_values = group['A'].unique().tolist()
    
    return columns, group_values

# 应用函数获取分组的列和组的值
result = grouped.apply(get_group_info)

# 打印结果
for index, (columns, group_values) in result.iteritems():
    print("Group: ", group_values)
    print("Columns: ", columns)
    print("--------------------")

Group:  ['bar']
Columns:  ['A', 'B', 'C', 'D']
--------------------
Group:  ['foo']
Columns:  ['A', 'B', 'C', 'D']
--------------------