single - python dataframe multiindex

Agregar un nivel a un pandas MultiIndex

Question

Tengo un DataFrame con un MultiIndex creado después de algunas agrupaciones:

import numpy as np
import pandas as p
from numpy.random import randn

df = p.DataFrame({
    ''A'' : [''a1'', ''a1'', ''a2'', ''a3'']
  , ''B'' : [''b1'', ''b2'', ''b3'', ''b4'']
  , ''Vals'' : randn(4)
}).groupby([''A'', ''B'']).sum()

df

Output>            Vals
Output> A  B           
Output> a1 b1 -1.632460
Output>    b2  0.596027
Output> a2 b3 -0.619130
Output> a3 b4 -0.002009

¿Cómo añado un nivel al MultiIndex para que lo convierta en algo así como:

Output>                       Vals
Output> FirstLevel A  B           
Output> Foo        a1 b1 -1.632460
Output>               b2  0.596027
Output>            a2 b3 -0.619130
Output>            a3 b4 -0.002009

Answer 1

Primero puede agregarlo como una columna normal y luego agregarlo al índice actual, así que:

df[''Firstlevel''] = ''Foo''
df.set_index(''Firstlevel'', append=True, inplace=True)

Y cambie el orden si es necesario con:

df.reorder_levels([''Firstlevel'', ''A'', ''B''])

Lo que resulta en:

                      Vals
Firstlevel A  B           
Foo        a1 b1  0.871563
              b2  0.494001
           a2 b3 -0.167811
           a3 b4 -1.353409

Answer 2

Una buena forma de hacerlo en una línea usando pandas.concat() :

import pandas as pd

pd.concat([df], keys=[''Foo''], names=[''Firstlevel''])

Esto se puede generalizar a muchos marcos de datos, vea los docs .