> Как выбрать строку с определённым индексом в DataFrame (Python)
Уровень: middle · Роль: backend · Категория: Технические вопросы
Компании: Сбер
Стек: Python
> Пример ответа
Для выбора строки по индексу в pandas DataFrame можно использовать метод .iloc[] (по числовому индексу) или .loc[] (по метке индекса).
Пример:
PYTHONimport pandas as pddf = 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...), и оба метода могут дать одинаковый результат, но их путать не стоит.
> Похожие задачи по backend
Как посчитать возраст в годах между текущей датой и датой рождения в Python
Какие есть ограничения на ключи в dict в Python?
Что такое range в Python и как он используется в качестве аргумента?
Какова временная сложность добавления элемента в список Python в конец, начало и середину
> ГОТОВЫ К СЛЕДУЮЩЕМУ СОБЕСЕДОВАНИЮ?
Запустите тренировочную сессию с ИИ и получите детальную обратную связь, чтобы увереннее проходить реальные интервью