python - 将包含两部分的列转换为 MultiIndex

Question

我有以下 DataFrame(从我无法更改的 CSV 文件中读取):

df = pd.DataFrame([['low', 0.5, 123, 0.8, 123], 
                   ['high', 0.7, 253, 0.9, 147]], 
                  columns=['type', 'g1 v0', 'g1 v1', 'g2 v0',  'g2 v1'])

我想在多重索引中“拆分”列 g1 v0、g1 v1 等，例如喜欢:

                   g1            g2
            v0     v1     v0     v1
   type     
0   low    0.5    123    0.8    123
1  high    0.7    253    0.9    147

基本上，我希望将列名的 gx/vy 部分放入两个单独的级别。

Answer 1

使用set_index + split :

df = df.set_index('type')
df.columns = df.columns.str.split(expand=True)
print (df)
       g1        g2     
       v0   v1   v0   v1
type                    
low   0.5  123  0.8  123
high  0.7  253  0.9  147

另一种解决方案 drop , split 和 concat :

df1 = df.drop('type', axis=1)
df1.columns = df1.columns.str.split(expand=True)
print (df1)
    g1        g2     
    v0   v1   v0   v1
0  0.5  123  0.8  123
1  0.7  253  0.9  147

df = pd.concat([df['type'].rename(('','type')), df1], axis=1)
print (df)
         g1        g2     
   type   v0   v1   v0   v1
0   low  0.5  123  0.8  123
1  high  0.7  253  0.9  147