Индекс набора панд с многоуровневыми столбцами - PullRequest
Новая
       24

Индекс набора панд с многоуровневыми столбцами

4 голосов
/ 27 июня 2019

рассмотрите ниже pd.DataFrame

df_index = pd.MultiIndex.from_product([['foo','bar'],['one','two','three']])
df = pd.DataFrame(np.random.randint(0,10,size=18, dtype='int').reshape((-1,6)), columns=df_index)

print(df)
                     foo                    bar
     one    two     three   one     two     three
   0    7   3         8       3     6         0
   1    2   5         9       4     3         6
   2    4   2         6       6     4         5

Я хочу установить 'foo' и все его подиндексы в качестве индекса. Как мне этого добиться? Я борюсь с 'set_index' и pd.IndexSlice, но все еще не могу найти решение

1 Ответ

1 голос
/ 27 июня 2019

Вам необходимо пройти все уровни MultiIndex как кортеж.Таким образом, правильный формат должен быть: 

df.set_index([('foo', 'one'), ('foo', 'two'), ('foo', 'three')])

Если это громоздко, вы можете создать свой индекс, используя представление списка, например: 

idx = [x for x in df.columns if x[0] == 'foo']
print(idx)
#  [('foo', 'one'), ('foo', 'two'), ('foo', 'three')]

df.set_index(idx)

[out] 

                                   bar          
                                   one two three
(foo, one) (foo, two) (foo, three)              
1          3          4              4   8     3
5          1          0              4   7     5
0          0          3              9   1     6
...