如何在熊猫中包含带有多索引过滤器的条件?

时间:2019-07-03 11:08:22

标签: python pandas multi-index

我有一个包含多索引的过滤器。我只想对该多索引中包含的值应用条件。有可能吗?

arrays = [np.array(['bar', 'bar', 'baz', 'baz', 'foo', 'foo', 'qux', 'qux']),
          np.array(['one', 'two', 'one', 'two', 'one', 'two', 'one', 'two'])]

df = pd.DataFrame(np.random.randn(8, 4), index=arrays)

# this multiindex comes from a model so it is fixed and we cannot know the values in advance
fixed_multiindex = pd.MultiIndex.from_tuples([('bar','one'), ('foo', 'one')])

df_multinidex_and_condition = df.loc[(fixed_multiindex) & (df[0] > -1.3)]

错误

Output error TypeError: other must be a MultiIndex or a list of tuples

1 个答案:

答案 0 :(得分:3)

使用Index.isin

df_multinidex_and_condition = df.loc[df.index.isin(fixed_multiindex) & (df[0] > -1.3)]
print (df_multinidex_and_condition)
                0         1         2         3
bar one -0.217679  0.821455  1.481278  1.331864
foo one  2.672172  0.464802  0.845930 -0.503542