獲得第一的K值的指數在混合指數熊貓系列

Question

L'ETS說我有熊貓系列稱爲Z：

index items 
155 14 
2  555 
1445 23 
855  3 

等。正如你所看到的，這些指數是混合的。我怎樣才能在不改變當前索引的情況下獲得K前幾行的索引列表？ 例如，對於K = 3，我想要：[155,2,1445]

謝謝。

Answer 1

您可以使用iloc與片：

s.iloc[:3].index 

或者使用head：

s.head(3).index 

---

s 
#index 
#155  14 
#2  555 
#1445  23 
#855  3 
#Name: items, dtype: int64 

s.iloc[:3].index.values 
#array([ 155, 2, 1445], dtype=int64) 

s.head(3).index.values 
#array([ 155, 2, 1445], dtype=int64) 

Answer 2

如果z是系列那麼只需要：

L = z.index[:3].tolist() 

編輯的評論：

看來你DataFrame和index是列：

print (type(z)) 
<class 'pandas.core.frame.DataFrame'> 

#set index with column index 
z = z.set_index('index') 
#select values by position and convert to list 
L = z.index[:3].tolist() 
print (L) 
[155, 2, 1445] 

如果索引列是可能的選擇列，然後獲得通過位置前N值與Series.iloc：

L = z['index'].iloc[:3].tolist() 

更好的是使用DataFrame.iloc，但需要positi在index柱也由get_loc的：

L = z.iloc[:3, df.columns.get_loc('index')].tolist()