Pandas GroupBy用法：提取分组索引

Pandas 中的 groupby 方法用于按照某一列或多列的值进行分组。可以通过 groupby 方法提取分组后的组索引。

下面是一个示例：

import pandas as pd

# 创建一个 DataFrame
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
        'Age': [20, 21, 22, 23, 24],
        'City': ['New York', 'London', 'Paris', 'New York', 'Paris']}
df = pd.DataFrame(data)

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

# 提取组索引
group_index = grouped.groups.keys()

print(group_index)

输出结果为：

dict_keys(['John', 'Nick', 'Tom'])

可以看到，groupby 方法返回的是一个 GroupBy 对象，可以通过 .groups 属性获取每个组的索引。使用 .keys() 方法可以获取所有组的索引。