我想向我的DF添加一个新级别（这样我就可以使用pd.reindex做其他事情）。我的DF基本有这样的东西：
df = pd.DataFrame({('A','a'): [-1,-1,0,10,12],
('A','b'): [0,1,2,3,-1],
('A','c'): [-1,1,0,10,12],
('A','d'): [1,1,2,3,-1],
('B','a'): [-20,-10,0,10,20],
('B','b'): [-200,-100,0,-1,200],
('B','c'): [-20,-10,0,10,20],
('B','d'): [-200,-100,0,100,200]
})
##df
A B
a b c d a b c d
0 -1 0 1 1 -20 -200 -20 -200
1 -1 1 -1 1 -10 -100 -10 -100
2 0 2 0 2 0 0 0 0
3 10 3 10 3 10 -1 10 100
4 12 -1 12 -1 20 200 20 200
我想分配新的级别键
L1 = a + b和L2 = c + d。我该怎么做呢？所需的输出将是
##df
A B
L1 L2 L1 L2
a b c d a b c d
0 -1 0 1 1 -20 -200 -20 -200
1 -1 1 -1 1 -10 -100 -10 -100
2 0 2 0 2 0 0 0 0
3 10 3 10 3 10 -1 10 100
4 12 -1 12 -1 20 200 20 200
编辑：目标是实现类似于here中要求的内容。这意味着某些行的同一KEY将具有
NA，这取决于其他列的值。例如。如果要通过分别测试列
a和c是否为负来过滤列b和d：##df
A B
L1 L2 L1 L2
a b c d a b c d
0 -1 0 1 1 NA NA NA NA
1 -1 1 -1 1 NA NA NA NA
2 0 2 0 2 0 0 0 0
3 10 3 10 3 NA NA 10 100
4 NA NA NA NA 20 200 20 200
最佳答案
您需要使用array创建新的map，然后分配：
d = {'a':'L1','b':'L1','c':'L2','d':'L2'}
a = df.columns.get_level_values(1).map(lambda x: d[x])
print (a)
['L1' 'L1' 'L2' 'L2' 'L1' 'L1' 'L2' 'L2']
df.columns = [df.columns.get_level_values(0),a,df.columns.get_level_values(1)]
#same as
df.columns = pd.MultiIndex.from_arrays([df.columns.get_level_values(0),
df.columns.get_level_values(1).map(lambda x: d[x]),
df.columns.get_level_values(1)])
print (df)
A B
L1 L2 L1 L2
a b c d a b c d
0 -1 0 -1 1 -20 -200 -20 -200
1 -1 1 1 1 -10 -100 -10 -100
2 0 2 0 2 0 0 0 0
3 10 3 10 3 10 -1 10 100
4 12 -1 12 -1 20 200 20 200
第二个输出确实很复杂，对我来说工作：
#filter columns
idx = pd.IndexSlice
mask = df.loc[:, idx[:,:,['b','d']]] < 0
print (mask)
A B
L1 L2 L1 L2
b d b d
0 False False True True
1 False False True True
2 False False False False
3 False False True False
4 True True False False
#create mask to columns a,c
mask1 = mask.reindex(columns=df.columns)
mask1 = mask1.groupby(level=[0,1], axis=1).apply(lambda x: x.bfill(axis=1))
print (mask1)
A B
L1 L2 L1 L2
a b c d a b c d
0 False False False False True True True True
1 False False False False True True True True
2 False False False False False False False False
3 False False False False True True False False
4 True True True True False False False False
print (df.mask(mask1))
A B
L1 L2 L1 L2
a b c d a b c d
0 -1.0 0.0 -1.0 1.0 NaN NaN NaN NaN
1 -1.0 1.0 1.0 1.0 NaN NaN NaN NaN
2 0.0 2.0 0.0 2.0 0.0 0.0 0.0 0.0
3 10.0 3.0 10.0 3.0 NaN NaN 10.0 100.0
4 NaN NaN NaN NaN 20.0 200.0 20.0 200.0
使用
reindex和method='bfill'的另一种解决方案，但必须进行两次转置（我认为这是错误-仅在MultiIndex中的index而不是MultiIndex中的columns起作用）：idx = pd.IndexSlice
mask = df.loc[:, idx[:,['b','d']]] < 0
print (mask)
A B
b d b d
0 False False True True
1 False False True True
2 False False False False
3 False False True False
4 True True False False
mask1 = mask.T.reindex(df.columns, method='bfill').T
print (mask1)
A B
a b c d a b c d
0 False False False False True True True True
1 False False False False True True True True
2 False False False False False False False False
3 False False False False True True False False
4 True True True True False False False False
print (df.mask(mask1))
A B
a b c d a b c d
0 -1.0 0.0 -1.0 1.0 NaN NaN NaN NaN
1 -1.0 1.0 1.0 1.0 NaN NaN NaN NaN
2 0.0 2.0 0.0 2.0 0.0 0.0 0.0 0.0
3 10.0 3.0 10.0 3.0 NaN NaN 10.0 100.0
4 NaN NaN NaN NaN 20.0 200.0 20.0 200.0
关于python - 在 Pandas 的DF中间添加等级，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/42487091/