🐼 Панды: выборка по условию

В библиотеке Pandas для нахождения строк по условию используйте метод df.loc[условие] или df[условие].

    
    import pandas as pd

    # Создаем DataFrame
    data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
            'Age': [25, 30, 35, 40],
            'Salary': [50000, 60000, 70000, 80000]}

    df = pd.DataFrame(data)

    # Выбираем строки, где возраст больше 30
    result = df.loc[df['Age'] > 30]  # или df[df['Age'] > 30]

    print(result)
    
    

В этом примере фильтруются строки, где возраст больше 30. Результат будет содержать только строки, где условие выполняется.

Условие where в библиотеке pandas

Библиотека pandas в Python предоставляет множество функций для анализа данных. Одна из таких функций - where. Давайте подробнее рассмотрим, как использовать условие where в библиотеке pandas.

1. Что такое условие where?

Условие where - это функция, которая позволяет фильтровать или заменять значения в объекте pandas на основе заданного условия. Она принимает два аргумента: условие и значение замены.

2. Как использовать условие where?

Для использования условия where необходимо импортировать библиотеку pandas:

import pandas as pd

Затем можно создать объект pandas, например, серию или датафрейм, и применить условие where к этому объекту.

Давайте рассмотрим пример использования условия where на серии pandas:

import pandas as pd

# Создание серии из списка
s = pd.Series([1, 2, 3, 4, 5])

# Фильтрация значений больше 3
filtered_s = s.where(s > 3, 0)

print(filtered_s)

В результате мы получим следующий вывод:

0    0
1    0
2    0
3    4
4    5
dtype: int64

Как видно из примера, значения, удовлетворяющие условию (больше 3), остаются неизменными, в то время как значения, не удовлетворяющие условию, заменяются на указанное значение (в данном примере 0).

Также условие where можно применять к датафреймам pandas. Давайте рассмотрим пример:

import pandas as pd

# Создание датафрейма из словаря
data = {'name': ['John', 'Alice', 'Bob', 'Charlie', 'David'],
        'age': [25, 30, 35, 40, 45],
        'salary': [50000, 60000, 70000, 80000, 90000]}

df = pd.DataFrame(data)

# Замена значений в столбце 'salary' на 0, если значение в столбце 'age' меньше 35
df['salary'] = df['salary'].where(df['age'] >= 35, 0)

print(df)

Результат будет следующим:

      name  age  salary
0     John   25       0
1    Alice   30       0
2      Bob   35   70000
3  Charlie   40   80000
4    David   45   90000

В данном примере мы заменили значения в столбце 'salary' на 0 только в тех случаях, когда значение в столбце 'age' было меньше 35.

3. Заключение

Условие where в библиотеке pandas является мощным инструментом для фильтрации и замены значений на основе заданного условия. Оно может быть применено как к сериям, так и к датафреймам. Надеюсь, что данная статья помогла вам понять, как использовать условие where в библиотеке pandas. Удачи в вашем дальнейшем изучении!