Как использовать метод pandas where isin для эффективной фильтрации данных?

Метод isin() в библиотеке pandas используется для фильтрации данных в DataFrame на основе заданных значений. Он принимает в качестве параметра набор значений и возвращает булеву маску, указывающую, где в DataFrame содержатся эти значения.

import pandas as pd

# Создание DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Paris', 'London', 'Tokyo']}
df = pd.DataFrame(data)

# Фильтрация данных, где 'City' содержит значения 'Paris' или 'London'
filtered_df = df[df['City'].isin(['Paris', 'London'])]

print(filtered_df)

В этом примере мы создаем DataFrame с данными о людях. Затем мы используем метод isin(), чтобы отфильтровать только те строки, где столбец 'City' содержит значения 'Paris' или 'London'. Результатом будет новый DataFrame, содержащий только эти строки.

Использование метода pandas where isin

Pandas - это мощная библиотека для анализа данных в Python. В ней содержатся различные методы, которые позволяют эффективно манипулировать и обрабатывать данные. Одним из таких методов является "where isin", который позволяет фильтровать данные, основываясь на определенных значениях в столбце. Давайте рассмотрим, как использовать этот метод.

Синтаксис метода where isin

Синтаксис метода where isin следующий:

df_filtered = df[df['column'].isin(values)]

Где:

• df - это объект DataFrame, с которым мы работаем;
• 'column' - это название столбца, в котором мы хотим выполнить фильтрацию;
• values - это список значений, по которым мы хотим выполнить фильтрацию.

Пример использования

Допустим, у нас есть следующий DataFrame:

import pandas as pd

data = {'Name': ['John', 'Jane', 'Adam', 'Emily'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'London', 'Paris', 'Tokyo']}

df = pd.DataFrame(data)

Хотим мы отфильтровать только те строки, где значение в столбце 'City' равно 'London' или 'Paris'. Вот как мы можем это сделать:

filtered_df = df[df['City'].isin(['London', 'Paris'])]

Результат:

   Name  Age    City
1  Jane   30  London
2  Adam   35   Paris

Мы получаем отфильтрованный DataFrame, в котором остаются только строки с городами Лондон и Париж.

Необходимость метода where isin

Метод where isin полезен, когда нам нужно выполнить фильтрацию на основе нескольких значений в столбце. Вместо того, чтобы использовать несколько логических операторов "или" для каждого значения, мы можем просто передать список значений в метод where isin и получить результирующий DataFrame. Это значительно сокращает количество кода и делает его более читаемым.

Вывод

Метод where isin является полезным инструментом для фильтрации данных в pandas. Он позволяет нам легко фильтровать данные на основе определенных значений в столбце, используя всего лишь несколько строк кода. Этот метод особенно полезен, когда нам требуется выполнить фильтрацию на основе нескольких значений в столбце. Используйте его в своих проектах и экспериментах с данными!