熊猫：如何创建一个多索引枢轴

Question

我有一套由两个变量定义的实验：scenario和height。对于每一个实验，我把3次测量：结果1，2和3 收集所有的结果数据框看起来是这样的：

import numpy as np 
import pandas as pd 

df = pd.DataFrame() 
df['Scenario']= np.repeat(['Scenario a','Scenario b','Scenario c'],3) 
df['height'] = np.tile([0,1,2],3) 
df['Result 1'] = np.arange(1,10) 
df['Result 2'] = np.arange(20,29) 
df['Result 3'] = np.arange(30,39) 

如果我运行以下命令：

mypiv = df.pivot('Scenario','height').transpose() 
writer = pd.ExcelWriter('test_df_pivot.xlsx') 
mypiv.to_excel(writer,'test df pivot') 
writer.save() 

我获得一个数据帧，其中列是scenarios，行有一个多指数result和height定义：

+----------+--------+------------+------------+------------+ 
|   | height | Scenario a | Scenario b | Scenario c | 
+----------+--------+------------+------------+------------+ 
| Result 1 |  0 |   1 |   4 |   7 | 
|   |  1 |   2 |   5 |   8 | 
|   |  2 |   3 |   6 |   9 | 
| Result 2 |  0 |   20 |   23 |   26 | 
|   |  1 |   21 |   24 |   27 | 
|   |  2 |   22 |   25 |   28 | 
| Result 3 |  0 |   30 |   33 |   36 | 
|   |  1 |   31 |   34 |   37 | 
|   |  2 |   32 |   35 |   38 | 
+----------+--------+------------+------------+------------+ 

如何创建索引交换的枢纽，即首先是height，然后是result？

我找不到直接创建它的方法。我设法得到我想要的交换水平和重新排序结果：

mypiv2 = mypiv.swaplevel(0,1 , axis=0).sortlevel(level=0,axis=0,sort_remaining=True) 

但我想知道是否有更直接的方法。

Answer 1

可以先set_index，然后用stackunstack：

print (df.set_index(['height','Scenario']).stack().unstack(level=1)) 
Scenario   Scenario a Scenario b Scenario c 
height            
0  Result 1   1   4   7 
     Result 2   20   23   26 
     Result 3   30   33   36 
1  Result 1   2   5   8 
     Result 2   21   24   27 
     Result 3   31   34   37 
2  Result 1   3   6   9 
     Result 2   22   25   28 
     Result 3   32   35   38