pandas groupby提取分组的列表

pandas的groupby方法可以用于按照某个列的值进行分组，并对每个分组进行操作。如果想要提取分组的列表，可以使用groupby的groups属性。

首先，使用groupby方法对DataFrame进行分组，然后通过groups属性获取分组的列表。groups属性返回一个字典，其中键是分组的值，值是该分组对应的行索引的列表。

以下是一个示例代码：

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]}
df = pd.DataFrame(data)

# 按照列A进行分组，并提取分组的列表
groups = df.groupby('A').groups
print(groups)

输出结果为：

{'bar': [1, 3, 5], 'foo': [0, 2, 4, 6, 7]}

可以看到，分组的列表以字典的形式返回，键是分组的值，值是该分组对应的行索引的列表。