分配给MultiIndex切片

时间:2014-02-03 22:34:57

标签: python pandas

In [332]: midx = MultiIndex(levels=[['zero', 'one'], ['x','y']],
   .....:                   labels=[[1,1,0,0],[1,0,1,0]])
   .....: 

In [333]: df = DataFrame(randn(4,2), index=midx)

In [334]: print(df)
               0         1
one  y  0.313092 -0.588491
     x  0.203166  1.632996
zero y -0.557549  0.126204
     x  1.643615 -0.067716

我想在第0列中将元素1 == y设置为nan。 使用.xs()我会这样做:df.xs('y', level=1)[0] = np.nan 这显然不起作用。 df.ix[(:, 'y'), 0] = np.nan的某些内容当然也是无效的。

2 个答案:

答案 0 :(得分:4)

In< = 0.13.1

In [51]: df = DataFrame(randn(4,2), index=midx)

In [52]: df.loc[df.index.get_loc_level('y',level=1)[0],0] = np.nan

In [53]: df
Out[53]: 
               0         1
one  y       NaN -0.260289
     x  0.122913  0.728180
zero y       NaN -0.010145
     x -0.532615  0.758914

[4 rows x 2 columns]

在0.14中,您能够使用新语法,请参阅:https://github.com/pydata/pandas/pull/6134

类似的东西:

df.loc[(slice(None),'y'),0] = np.nan

应该有用。

答案 1 :(得分:3)

关于改进MultiIndex切片语法有一个开放的Github issue,但是现在我建议:

In [59]: idx_slice = [(x, 'y') for x in df.index.levels[0]]

In [60]: idx_slice
Out[60]: [('zero', 'y'), ('one', 'y')]

In [62]: df.ix[idx_slice, 0] = np.nan

In [63]: df
Out[63]: 
               0         1
one  y       NaN  0.210371
     x -1.109476  1.861331
zero y       NaN  0.189710
     x -1.013922 -1.465135

[4 rows x 2 columns]

希望这适用于您的真实数据集!