Complex pandas example

pandas_complex_filter.py

import pandas as pd

# Создаем пример DataFrame
data = {
    'Name': ['John', 'Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35, 40],
    'Country': ['USA', 'Canada', 'UK', 'Australia']
}
df = pd.DataFrame(data)

# Функция для авторазметки
def label(row):
    if row['Age'] < 30 and row['Country'] == 'USA':
        return 'Young American'
    elif row['Age'] >= 30 and row['Country'] == 'USA':
        return 'Adult American'
    elif row['Age'] < 30 and row['Country'] != 'USA':
        return 'Young Non-American'
    elif row['Age'] >= 30 and row['Country'] != 'USA':
        return 'Adult Non-American'
    else:
        return 'Unknown'

# Применяем функцию авторазметки к DataFrame
df['Label'] = df.apply(label, axis=1)

# Выводим результат
print(df)