Uzyskiwanie indeksu liczb całkowitych wiersza Pandas DataFrame spełniającego warunek?

Question

Mam następujący DataFrame:

a b c 
b 
2 1 2 3 
5 4 5 6 

Jak widać, kolumna b jest używany jako wskaźnik. Chcę uzyskać numer porządkowy rzędu spełniającego ('b' == 5), który w tym przypadku byłby 1.

Testowana kolumna może być kolumną indeksową (jak w przypadku b w tym przypadku) lub zwykłą kolumną, np. Może chciałbym znaleźć indeks wiersza spełniający ('c' == 6).

Answer 1

Można użyć np.where tak:

import pandas as pd 
import numpy as np 
df = pd.DataFrame(np.arange(1,7).reshape(2,3), 
        columns = list('abc'), 
        index=pd.Series([2,5], name='b')) 
print(df) 
# a b c 
# b   
# 2 1 2 3 
# 5 4 5 6 
print(np.where(df.index==5)[0]) 
# [1] 
print(np.where(df['c']==6)[0]) 
# [1] 

Wartość zwrócona jest tablicą, ponieważ nie może być więcej niż jeden wiersz w szczególności indeks lub wartość w kolumnie.

Answer 2

Zamiast tego należy użyć Index.get_loc.

Ponownie wykorzystując skonfigurowany kod @ unutbu, uzyskasz identyczne wyniki.

>>> import pandas as pd 
>>> import numpy as np 


>>> df = pd.DataFrame(np.arange(1,7).reshape(2,3), 
        columns = list('abc'), 
        index=pd.Series([2,5], name='b')) 
>>> df 
    a b c 
b 
2 1 2 3 
5 4 5 6 
>>> df.index.get_loc(5) 
1