熊貓MultiIndex獲取標籤值爲

Question

的所有行假設您擁有帶MultiIndex的Panda DataFrame。您想要獲取具有特定值的標籤的所有行。你怎麼做到這一點？

我首先想到的是一個布爾面具...

df[df.index.labels == 1].head()

但這不起作用。

謝謝！

Answer 1

您需要指定您使用的索引。在我的例子中，我採取了第二個指標（我的數據幀是是因爲它是如此的大熊貓的多指標頁）：

s[s.index.labels[1]==1] 

實際上，你可以看到指數是如何構建的，如果你鍵入：

s.index 

所形成的結構是：

MultiIndex(levels=[['bar', 'baz', 'foo', 'qux'], [1, 2]], 
     labels=[[0, 0, 1, 1, 2, 2, 3, 3], [0, 1, 0, 1, 0, 1, 0, 1]], 
     names=['first', 'second']) 

下面我有全碼：

>>> import pandas as pd 
>>> import numpy as np 
>>> arrays = [['bar', 'bar', 'baz', 'baz', 'foo', 'foo', 'qux', 'qux'], 
...   [1, 2, 1, 2, 1, 2, 1, 2]] 
... 
>>> tuples = list(zip(*arrays)) 
>>> index = pd.MultiIndex.from_tuples(tuples, names=['first', 'second']) 
>>> s = pd.Series(np.random.randn(8), index=index) 
>>> s[s.index.labels[1]==1] 
first second 
bar 2  -0.304029 
baz 2  -1.216370 
foo 2   1.401905 
qux 2  -0.411468 
dtype: float64 

Answer 2

我會用xs (cross-section)：

In [11]: df = pd.DataFrame([[1, 2, 3], [3, 4, 5]], columns=list("ABC")).set_index(["A", "B"]) 

In [12]: df 
Out[12]: 
    C 
A B 
1 2 3 
3 4 5 

那麼你可以採取那些具有A級等於1：

In [13]: df.xs(key=1, level="A") 
Out[13]: 
    C 
B 
2 3 

使用drop_level=False做的過濾器（不刪除的A股指數）：

In [14]: df.xs(key=1, level="A", drop_level=False) 
Out[14]: 
    C 
A B 
1 2 3 

Answer 3

替代解決方案：

In [62]: df = pd.DataFrame({'idx1': ['A','B','C'], 'idx2':[1,2,3], 'val': [30,10,20]}).set_index(['idx1','idx2']) 

In [63]: df 
Out[63]: 
      val 
idx1 idx2 
A 1  30 
B 2  10 
C 3  20 

In [64]: df[df.index.get_level_values('idx2') == 2] 
Out[64]: 
      val 
idx1 idx2 
B 2  10 

In [65]: df[df.index.get_level_values(1) == 2] 
Out[65]: 
      val 
idx1 idx2 
B 2  10