如何向 MultiIndex 添加级别？答案

【问题标题】：How can I add a level to a MultiIndex?如何向 MultiIndex 添加级别？
【发布时间】：2017-07-25 12:46:51
【问题描述】：

index = [np.array(['foo', 'foo', 'qux']),
         np.array(['a', 'b', 'a'])]
data = np.random.randn(3, 2)
columns = ["X", "Y"]
df = pd.DataFrame(data, index=index, columns=columns)
df.index.names = ["Level0", "Level1"]
print df

                      X         Y
Level0 Level1                    
foo    a       0.418549  0.252685
       b      -1.307099  0.202833
qux    a       0.046095 -0.968976

新关卡

我想获取 df 的索引并创建一个新的 MultiIndex，它现在有一个额外的级别。

new_level_name = "New level"
new_level_labels = ['p', 'q']
# new_multi-index

所需的多索引

Level0 Level1 Level2                    
foo    a      p       
              q       
       b      p      
              q       
qux    a      p      
              q

【问题讨论】：

标签： python pandas dataframe multi-index

【解决方案1】：

最简单的方法是将stack 与新的DataFrame 一起使用，并按新的级别值列：

df1 = pd.DataFrame(data=1,index=df.index, columns=new_level_labels).stack()
df1.index.names = ['Level0','Level1',new_level_name]
print (df1)
Level0  Level1  New level
foo     a       p            1
                q            1
        b       p            1
                q            1
qux     a       p            1
                q            1
dtype: int64

print (df1.index)
MultiIndex(levels=[['foo', 'qux'], ['a', 'b'], ['p', 'q']],
           labels=[[0, 0, 0, 0, 1, 1], [0, 0, 1, 1, 0, 0], [0, 1, 0, 1, 0, 1]],
           names=['Level0', 'Level1', 'New level'])

【讨论】：

啊好极了！我尝试了几乎完全相同的事情，但没有包含 data=1！
是的，我也尝试过，但如果不是1，则默认为NaN，并且所有行都被删除。
结果实际上是一个 pd.Series，而不是 DataFrame (?)。不敢相信没有更好的方法...
@n1000 - 你觉得print (df.reindex(df1.index)) 吗？

【解决方案2】：

不完全确定您想要的 DataFrame 中应该包含哪些数据，但对于类似的问题，我使用了melt()：

df1 = df.reset_index()
pd.melt(df1, id_vars=['Level0', 'Level1'])


    Level0  Level1  variable    value
0   foo a   X   0.678564
1   foo b   X   -0.609134
2   qux a   X   0.505178
3   foo a   Y   -2.593380
4   foo b   Y   -0.232796
5   qux a   Y   -1.420875

【讨论】：