> Как выбрать строку с определённым индексом в DataFrame (Python)

Уровень: middle · Роль: backend · Категория: Технические вопросы

Компании: Сбер

Стек: Python

> Пример ответа

Для выбора строки по индексу в pandas DataFrame можно использовать метод .iloc[] (по числовому индексу) или .loc[] (по метке индекса).

Пример:

PYTHON
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]}, index=['x', 'y', 'z'])
# По числовому индексу (позиции) - строка с индексом 1 (вторая строка)
row_iloc = df.iloc[1] # Результат: A=2, B=5
# По метке индекса - строка с меткой 'y'
row_loc = df.loc['y'] # Результат: A=2, B=5

Если нужно выбрать несколько строк, передайте список или срез:

PYTHON
# Несколько строк по позициям
rows = df.iloc[[0, 2]] # Первая и третья строки
# Срез по позициям
rows_slice = df.iloc[0:2] # Первые две строки

Важно: .iloc использует целочисленные позиции (начиная с 0), а .loc - метки индекса. Если индекс не задан явно, pandas присваивает целочисленные метки (0, 1, 2...), и оба метода могут дать одинаковый результат, но их путать не стоит.

> ГОТОВЫ К СЛЕДУЮЩЕМУ СОБЕСЕДОВАНИЮ?

Запустите тренировочную сессию с ИИ и получите детальную обратную связь, чтобы увереннее проходить реальные интервью