pandas groupby后获取索引

在Pandas中，可以使用`groupby`方法按照指定的列或函数分组，并对每个分组进行操作。如果想要获取分组后的索引，可以使用`groupby`方法后跟`indices`属性。该属性返回一个字典，其中键是每个分组的名称，值是一个包含该分组索引的数组。以下是一个示例：

import pandas as pd

# 创建一个DataFrame
df = pd.DataFrame({
    'key': ['A', 'B', 'C', 'A', 'B', 'C'],
    'value': [1, 2, 3, 4, 5, 6]
})

# 按照key列分组并获取索引
grouped = df.groupby('key')
indices = grouped.indices
print(indices)

输出结果为：

{'A': array([0, 3]), 'B': array([1, 4]), 'C': array([2, 5])}

可以看到，返回的字典中，键为'A'、'B'、'C'，对应分组的名称，值为包含该分组索引的数组。