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()方法可以获取所有组的索引。