Python 3.8.8
使用多重索引来进行三维面板操作
import pandas as pd
import numpy as np
df = pd.DataFrame({'class':['A','A','A','B','B','B','C','C','D'],
'id':['a','b','c','a','b','c','a','b','d'],
'value':[1,2,3,4,5,6,7,8,9]})
df.set_index(['class', 'id'], inplace=True) # 设置多重索引
# df.index.names = ['class', 'id'] # 为行多重索引命名
temp = df.loc[(slice(None),'b'),:] # 切片筛选第二索引的'b'
# temp2 = df.query("id == 'b'")
df.sort_index(level='id', inplace=True) # 按指定的行索引排序 若未指定 则按行索引的顺寻排列
df = df.swaplevel('class', 'id') # 更改多索引行的先后顺序
# df.reset_index(level='class', inplace=True) # 从多索引中释放class的索引
print(df.index.is_lexsorted()) # 检查index是否有序 方便切片筛选index
1st_index = df.index.levels[0].to_list() # 提取unique的第一索引
参考链接:
1.在pandas多重索引multiIndex中选定指定索引的行
2.创建多层行索引、多层索引对象的索引与切片操作