Crie uma nova coluna salary_category no DataFrame com base no valor do salário (alto/baixo)

Чтобы создать новый столбец salary_category в DataFrame на основе значения зарплаты, можно использовать условное присваивание с помощью метода apply или np.where в pandas.

Пример с использованием np.where:

import pandas as pd
import numpy as np

# Пример DataFrame
df = pd.DataFrame({
    'salary': [50000, 120000, 70000, 150000]
})

# Определяем порог для разделения на high и low
threshold = 100000

# Создаем новый столбец salary_category
df['salary_category'] = np.where(df['salary'] >= threshold, 'high', 'low')

print(df)

В результате столбец salary_category будет содержать значения 'high' для зарплат выше или равных 100000 и 'low' для остальных.