仅给定级别 0 列选择器和对齐级别 1 的新 MultiIndex 列的分配

Question

我想向多索引列数据框的 level(0) 添加一列，该列与其他列具有相同的结构，即：level(1) 索引相同：

In [151]: df
Out[151]: 
first        bar                 baz           
second       one       two       one       two 
A       0.487880 -0.487661 -1.030176  0.100813 
B       0.267913  1.918923  0.132791  0.178503
C       1.550526 -0.312235 -1.177689 -0.081596 

虽然我可以进行正常操作并获得与 df 的 level(1) 对齐的单级列索引数据框：

In [152]: df['bar'] - df['baz']

我无法执行普通的列分配 level(0) 来创建 level(0) 列，其 level(1) 由生成的（对齐的）数据框填充：

In [153]: df['foo'] = df['bar'] - df['baz']

这是怎么做到的？

Answer 1

我认为我们需要在结果中创建多个index，然后返回join

s = df['bar'] - df['baz']
s.columns = pd.MultiIndex.from_product([['foo'],s.columns])
df = df.join(s)
df
             bar                 baz                 foo          
             one       two       one       two       one       two
second                                                            
A       0.487880 -0.487661 -1.030176  0.100813  1.518056 -0.588474
B       0.267913  1.918923  0.132791  0.178503  0.135122  1.740420
C       1.550526 -0.312235 -1.177689 -0.081596  2.728215 -0.230639