Отбор строк из DataFrame по значениям в столбце

Проблема

При работе с данными на языке программирования Python и библиотекой Pandas часто возникает задача отбора определенных строк из DataFrame на основе значений в одном или нескольких столбцах.

Представим, что имеется DataFrame, содержащий информацию о студентах и их оценках по различным предметам:

import pandas as pd

data = {
    'name': ['Alex', 'John', 'Kate', 'Sam', 'Alice'],
    'math': [5, 4, 4, 5, 3],
    'physics': [4, 5, 5, 4, 3],
    'chemistry': [5, 4, 3, 4, 3]
}
df = pd.DataFrame(data)

Требуется выбрать всех студентов, у которых оценка по математике равна 5.

Решение

Для решения задачи в Python с использованием библиотеки Pandas можно использовать метод .loc[], который предназначен для доступа к группе строк и столбцов по метке или массиву булевых значений.

Для отбора строк, в которых значение в столбце ‘math’ равно 5, можно использовать следующий код:

selected_students = df.loc[df['math'] == 5]

В данном случае df['math'] == 5 создает массив булевых значений (True или False) для каждой строки DataFrame. Если значение в столбце ‘math’ равно 5, то соответствующее значение в массиве будет True, в противном случае — False. Затем метод .loc[] выбирает только те строки, для которых соответствующее значение в массиве равно True.

Выбранные строки можно вывести на экран с помощью команды print(selected_students).

Курсы по программированию

Помогаем получить новую профессию с гарантией трудоустройства!

Подробнее

Заключение

Python и библиотека Pandas предоставляют удобные и мощные инструменты для работы с данными, включая отбор строк из DataFrame по значениям в определенном столбце. Важно знать и понимать, как использовать эти инструменты, чтобы эффективно решать поставленные задачи.

Тест на профориентацию

За 10 минут узнайте, как ваш опыт пригодиться в IT индустрии

Подробнее