Сортировка столбцов мультииндексного фрейма данных - PullRequest
Новая
       10

Сортировка столбцов мультииндексного фрейма данных

2 голосов
/ 10 июня 2019

У меня очень большой мультииндексный фрейм данных с около 500 столбцами, и у каждого столбца есть 2 вложенных столбца.

Фрейм данных df выглядит как: 

                  B2                  B5             B3
bkt              A1      A2           A2      A1     Z2      C1
Date                                                                        
2019-06-11       0.8     0.2          -6.0    -0.8   -4.1    -0.6    
2019-06-12       0.8     0.2          -6.9    -1.6   -5.3    -1.2    

df.columns
MultiIndex(levels=[['B2', 'B5', 'B3', .....], ['A1', 'A2' ......]],
           labels=[[1, 1, ....], [1, 0, ....]],
           names=[None, 'bkt'])

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

                 B2                  B3             B5
bkt              A1      A2          C1      Z2     A1      A2
Date                                                                        
2019-06-11       ..
2019-06-12       ..

.. представляет значения из исходного кадра данных. Я просто не перепечатывал их.

Настройка 

df = pd.DataFrame([
    [.8, .2, -6., -.8, -4.1, -.6],
    [.8, .2, -6.9, -1.6, -5.3, -1.2]
],
    pd.date_range('2019-06-11', periods=2, name='Date'),
    pd.MultiIndex.from_arrays([
        'B2 B2 B5 B5 B3 B3'.split(),
        'A1 A2 A2 A1 Z2 C1'.split()
    ], names=[None, 'bkt'])
)

1 Ответ

3 голосов
/ 10 июня 2019

Используя sort_index и назначить его обратно 

df.columns=df.sort_index(axis=1,level=[0,1],ascending=[True,False]).columns

А из piR нам не нужно создавать копию df, просто сделайте модификацию со столбцами 

df.columns=df.columns.sort_values(ascending=[True, False])
...