将Pandas DataFrame转换为分层列/更改列层次结构

时间:2015-03-02 20:47:18

标签: python pandas pivot

我想转动数据框,如:

       dim1   Value_V     Value_y   instance
0      A_1     50.000000        0   instance200
1      A_2   6500.000000        1   instance200
2      A_3     50.000000        0   instance200
3      A_4   4305.922313        1   instance200

进入具有以下分层列的数据框:

              A_1               A_2               A_3                .....
              Value_V  Value_y  Value_V  Value_y  Value_V  Value_y
instance200   50       0        6500     1        50       0

我试过了df = df.pivot(index = "instance", columns = "dim1"),但它只会给我一个这样的框架:

              Value_V               Value_y                              
              A_1   A_2   A_3 ....  A_1  A_2  A_3 ....
instance200   50    6500  50        0    1    0

如何更改列的层次结构?

2 个答案:

答案 0 :(得分:4)

我自己想出来了:

df = df.swaplevel(0,1,axis = 1).sort(axis = 1)

会做

答案 1 :(得分:2)

您需要的是 reorder_levels ,然后对列进行排序,如下所示:

import pandas as pd

df = pd.read_clipboard()

df
Out[8]:
dim1    Value_V Value_y instance
0   A_1 50.000000   0   instance200
1   A_2 6500.000000 1   instance200
2   A_3 50.000000   0   instance200
3   A_4 4305.922313 1   instance200
In [9]:

df.pivot('instance', 'dim1').reorder_levels([1, 0], axis=1).sort(axis=1)
Out[9]:
dim1        A_1             A_2             A_3             A_4
            Value_V Value_y Value_V Value_y Value_V Value_y Value_V Value_y
instance                                
instance200 50      0       6500    1       50      0       4305.922313 1