🔍 Как найти индекс строки в Pandas: простые способы и советы

import pandas as pd

# Создаем DataFrame для примера
data = {'Name': ['John', 'Jane', 'Alex', 'Sarah'],
        'Age': [25, 30, 28, 35],
        'City': ['New York', 'London', 'Paris', 'Sydney']}

df = pd.DataFrame(data)

# Найти индекс строки, где значение в столбце 'Name' равно 'Alex'
index = df[df['Name'] == 'Alex'].index[0]
print(index)
    

Как найти индекс строки в pandas

В библиотеке pandas для работы с данными в таблицах мы часто сталкиваемся с задачей поиска индекса строки. Индекс строки является уникальным идентификатором каждой строки в таблице. Он может быть полезен при фильтрации данных или выполнении операций с определенными строками.

Метод `index`

Один из способов найти индекс строки в pandas - использовать метод `index`. Этот метод возвращает индекс первого вхождения значения в таблице.

import pandas as pd

# Создание таблицы
data = {'Имя': ['Анна', 'Иван', 'Мария', 'Алексей'],
        'Возраст': [25, 30, 28, 35],
        'Город': ['Москва', 'Санкт-Петербург', 'Казань', 'Екатеринбург']}

df = pd.DataFrame(data)

# Нахождение индекса строки по значению
index = df[df['Имя'] == 'Иван'].index[0]
print(index)
    

В данном примере мы создаем таблицу `df` с данными, а затем с помощью фильтрации по колонке "Имя" находим индекс строки, содержащей значение "Иван". Результат будет выведен на экран. Обратите внимание, что используется индекс `[0]`, так как метод `index` возвращает список индексов, и мы извлекаем первый элемент.

Метод `get_loc`

Другой способ найти индекс строки - использовать метод `get_loc`. Этот метод возвращает индекс строки, если его значение находится в индексе таблицы.

import pandas as pd

# Создание таблицы
data = {'Имя': ['Анна', 'Иван', 'Мария', 'Алексей'],
        'Возраст': [25, 30, 28, 35],
        'Город': ['Москва', 'Санкт-Петербург', 'Казань', 'Екатеринбург']}

df = pd.DataFrame(data)
df.set_index('Имя', inplace=True)  # Установка колонки "Имя" в качестве индекса

# Нахождение индекса строки по значению
index = df.index.get_loc('Мария')
print(index)
    

В этом примере мы устанавливаем колонку "Имя" в качестве индекса с помощью метода `set_index`. Затем мы используем метод `get_loc`, чтобы найти индекс строки с именем "Мария". Результат будет выведен на экран.

Метод `loc`

Третий способ найти индекс строки - использовать метод `loc`. Этот метод позволяет обращаться к строкам по индексу или условию и выполнять операции над ними.

import pandas as pd

# Создание таблицы
data = {'Имя': ['Анна', 'Иван', 'Мария', 'Алексей'],
        'Возраст': [25, 30, 28, 35],
        'Город': ['Москва', 'Санкт-Петербург', 'Казань', 'Екатеринбург']}

df = pd.DataFrame(data)
df.set_index('Имя', inplace=True)  # Установка колонки "Имя" в качестве индекса

# Нахождение индекса строки по значению
index = df.loc[df['Возраст'] == 30].index[0]
print(index)
    

Здесь мы также устанавливаем колонку "Имя" в качестве индекса, а затем с помощью метода `loc` фильтруем строки по условию ("Возраст" равен 30). Получаем индекс первой строки, удовлетворяющей условию.

Вывод

Вы научились находить индекс строки в pandas с помощью методов `index`, `get_loc` и `loc`. Знание этих методов поможет вам более эффективно работать с данными в таблицах и выполнять различные операции поиска и фильтрации.

Успешного использования pandas!