import pandas as pd import numpy as np
data = {'group': ['A', 'A', 'B', 'B', 'C', 'C'], 'value': [1, 2, 3, 4, 5, 6]} df = pd.DataFrame(data)
grouped = df.groupby('group') grouped_index = grouped.apply(lambda x: x.set_index('group'))
result = grouped_index.loc[['A', 'C']]
输出结果:
value
group
A 1
A 2
C 5
C 6
上一篇:按索引列表删除字典元素
下一篇:按索引排除值来筛选列表