Создать два уровня мультииндекс - PullRequest
Новая
       9

Создать два уровня мультииндекс

3 голосов
/ 11 апреля 2020

Я могу создать df и затем изменить его, чтобы он имел двухуровневый индекс столбца следующим образом: 

import pandas as pd
import numpy as np


idx = pd.MultiIndex.from_product([['bar', 'baz', 'foo', 'qux'], ['one', 'two', 'three']])

df = pd.DataFrame(np.random.randn(12, 6), index=idx, columns=['C', 'D', 'E', 'F', 'G', 'H'])
print(df, '\n')

df.columns = pd.MultiIndex.from_product([['A'], df.columns])
print(df.head(3))

Я получаю это: 

                  A                                                  
                  C         D         E         F         G         H
bar one   -0.370228  1.246188  0.673553  0.116890  0.129511  0.126562
    two   -1.059752 -0.357985 -0.189913  1.080814  0.588176  0.212053
    three -0.345277 -1.227097  0.915477  1.475285 -1.342885  0.149785

Но я хочу вот что : 

                  A                             B                      
                  C         D         E         F         G         H
bar one   -0.370228  1.246188  0.673553  0.116890  0.129511  0.126562
    two   -1.059752 -0.357985 -0.189913  1.080814  0.588176  0.212053
    three -0.345277 -1.227097  0.915477  1.475285 -1.342885  0.149785

Чтобы к моим столбцам обращались как: A C, AD, AE, BF, BG, BH

Я пробовал это (и несколько других вещей): 

df.columns = pd.MultiIndex.from_product([[['A'], df[['C', 'D', 'E']]], [['B'], df[['F', 'G', 'H']]]])

Но я продолжаю получать эту ошибку: 

TypeError: unhashable type: 'list'

Как я могу создать мультииндекс так, как я желаю?

1 Ответ

0 голосов
/ 11 апреля 2020

Вы просто назначаете двумерный массив в качестве столбцов информационного кадра: 

df.columns = [
    ['A']*3 + ['B']*3,
    ['C', 'D', 'E', 'F', 'G', 'H']
]

Результат: 

                  A                             B                    
                  C         D         E         F         G         H
bar one   -0.682664  0.221460 -0.595096 -0.327291  0.997758 -0.100819
    two    0.698674 -0.433741  1.572080  0.849064  0.666364 -0.313989
    three -0.597167  0.322044 -0.693410  1.551602  0.718136 -0.881298
baz one    0.701170 -0.164332  0.111689  0.192714 -0.743848 -1.447928
    two    0.504745 -0.059556 -1.673365  0.734746  0.586669  1.541826
...