Pandas 在 MultiIndex DataFrame 中选择特定的较低级别的列

Question

我不知道这个问题是否再次得到解答，但我还没有找到类似的东西

我有一个包含两级列的多索引数据帧，例如：

arrays = [np.array(['bar', 'bar','bar', 'foo', 'foo','foo', 'qux', 'qux', 'qux']),
np.array(['one', 'two', 'three', 'one', 'two', 'three', 'one', 'two','three'])]
df = pd.DataFrame(np.random.randn(3, 9), columns=arrays)
print(df)
bar                           foo                           qux  
one       two     three       one       two     three       one   
0  1.255724 -0.692387 -1.485324  2.265736  0.494645  1.973369 -0.326260   
1 -0.903874  0.695460 -0.950076  0.181590 -2.345611  1.288061  0.980166   
2 -0.294882  1.034745  1.423288 -0.895625 -0.847338  0.470444  0.373579   

two     three  
0  0.136427 -0.136479  
1  0.702732 -1.894376  
2  0.506240 -0.456519  

我想为每个第一级列单独从第二级中选择特定列。

例如，我想得到这样的结果：

bar                 foo       qux          
one       two       two       one     three
0  1.255724 -0.692387  0.494645 -0.326260 -0.136479
1 -0.903874  0.695460 -2.345611  0.980166 -1.894376
2 -0.294882  1.034745 -0.847338  0.373579 -0.456519

我已经看到了这个问题，但这不是我想要实现的目标。

现在我这样做：

level0 = ['bar','foo','qux']
level1 = [['one','two'],['two'],['one','three']]
df_list=[]
for i,value in enumerate(level0):
df_list.append(df.loc[:,(value,level1[i])])
new_df = pd.concat([i for i in df_list],axis=1)
print(new_df)

但在我看来，这不是最好的解决方案。

有没有更好的(更"熊猫"(方法来解决这个问题？

Answer 1

您可以先挑选列并使用列提取，而不是连接数据;

cols = pd.concat([pd.DataFrame({'level_0':x, 'level_1':y}) 
for x,y in zip(level0,level1)]
).values
df[cols]

输出：

bar                 foo       qux          
one       two       two       one     three
0  0.729061 -0.876547  0.312557  0.736568  0.250469
1  0.619194  0.451023  0.803252 -1.636403 -0.854607
2  0.254690 -1.054859 -1.223274  0.398411 -1.448396