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]}

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