как выбрать строку в pandas

Аватар пользователя Maksim Litvinov
Maksim Litvinov
11 сентября 2024

Для выбора определенной строки в DataFrame библиотеки Pandas есть несколько способов, в зависимости от поставленной задачи. Ниже несколько примеров, как можно это сделать

  • Использование метода iloc для выбора строки по ее числовому индексу:
import pandas as pd

# Создание DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# Выбор строки с индексом 0
row = df.iloc[0]
print(row)
  • Использование метода loc для выбора строки по ее метке индекса:
import pandas as pd

# Создание DataFrame с метками индекса
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]}, index=['row1', 'row2', 'row3'])

# Выбор строки с меткой 'row1'
row = df.loc['row1']
print(row)
  • Использование условной индексации для выбора строки по условию:
import pandas as pd

# Создание DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})

# Выбор строки, где значение в столбце 'A' равно 2
row = df[df['A'] == 2]
print(row)
1 0