为什么pandas多索引数据帧切片似乎不一致?

时间:2016-08-07 16:54:15

标签: python pandas numpy dataframe multi-index

为什么在切割多索引数据帧时,只要切换0级索引,就可以使用更简单的语法?以下是一个示例数据框:

           hi
a b   c      
1 foo baz   0
      can   1
  bar baz   2
      can   3
2 foo baz   4
      can   5
  bar baz   6
      can   7
3 foo baz   8
      can   9
  bar baz  10
      can  11

这些工作:

df.loc[1, 'foo', :]
df.loc[1, :, 'can']

虽然这不是:

df.loc[:, 'foo', 'can']

强迫我改用其中一个:

df.loc[(slice(None), 'foo', 'can'), :]
df.loc[pd.IndexSlice[:, 'foo', 'can'], :]

以下是相同的示例,但更详细:

In [1]: import pandas as pd
import numpy as np

ix = pd.MultiIndex.from_product([[1, 2, 3], ['foo', 'bar'], ['baz', 'can']], names=['a', 'b', 'c'])
data = np.arange(len(ix))
df = pd.DataFrame(data, index=ix, columns=['hi'])
print df

           hi
a b   c      
1 foo baz   0
      can   1
  bar baz   2
      can   3
2 foo baz   4
      can   5
  bar baz   6
      can   7
3 foo baz   8
      can   9
  bar baz  10
      can  11

In [2]: df.sort_index(inplace=True)
print df.loc[1, 'foo', :]

           hi
a b   c      
1 foo baz   0
      can   1

In [3]: print df.loc[1, :, 'can']

           hi
a b   c      
1 bar can   3
  foo can   1

In [4]: print df.loc[:, 'foo', 'can']

KeyError: 'the label [foo] is not in the [columns]'

In [5]: print df.loc[(slice(None), 'foo', 'can'), :]

           hi
a b   c      
1 foo can   1
2 foo can   5
3 foo can   9

In [6]: print df.loc[pd.IndexSlice[:, 'foo', 'can'], :]

           hi
a b   c      
1 foo can   1
2 foo can   5
3 foo can   9

1 个答案:

答案 0 :(得分:1)

这三个例子在技术上都是模棱两可的,它只是前两个,pandas正确猜出了你的意图。由于切片行,选择列(即df.loc[:, columns]是一种常见的习语,推理似乎选择了这种解释。

推理有点混乱,所以我觉得明白要好得多。如果别名IndexSlice

,那就不是那么多额外打字了
idx = pd.IndexSlice
df.loc[idx[1, 'foo'], :]
df.loc[idx[1, :, 'can'], :]
df.loc[idx[:, 'foo', 'can'], :]