Блог

Как в Python Pandas найти числовой индекс фрейма данных?

Рассмотрим следующий пример.

 import pandas as pd

df = pd.DataFrame(np.arange(20).reshape(5,4), index = np.arange(2,7),columns=["A","B","C","D"])

Output:

    A   B   C   D
2   0   1   2   3
3   4   5   6   7
4   8   9  10  11
5  12  13  14  15
6  16  17  18  19
  

Как вы находите индекс местоположения df.index == 4. Ответ должен быть 2.

Попробуйте .get_loc

 In [4]: df.index.get_loc(4)
Out[4]: 2
  

Проверьте get_indexer , обратите внимание, что разница здесь заключается get_loc в том, что будет выполняться одно значение для каждого вызываемого, get_indexer вы можете передать список

 df.index.get_indexer([4])
array([2], dtype=int64)