Comment créer une nouvelle colonne dans Pandas basée sur une logique conditionnelle sur plusieurs colonnes ?

Créer une nouvelle colonne basée sur des critères conditionnels de plusieurs colonnes

Pour créer une nouvelle colonne basée sur des critères conditionnels de plusieurs colonnes, appliquez votre personnalisation fonction sur toutes les lignes de votre DataFrame à l'aide de la méthode apply().

import pandas as pd

def label_race(row):
    if row['eri_hispanic'] == 1:
        return 'Hispanic'
    elif row['eri_afr_amer'] + row['eri_asian'] + row['eri_hawaiian'] + row['eri_nat_amer'] + row['eri_white'] > 1:
        return 'Two Or More'
    elif row['eri_nat_amer'] == 1:
        return 'A/I AK Native'
    elif row['eri_asian'] == 1:
        return 'Asian'
    elif row['eri_afr_amer'] == 1:
        return 'Black/AA'
    elif row['eri_hawaiian'] == 1:
        return 'Haw/Pac Isl.'
    elif row['eri_white'] == 1:
        return 'White'
    else:
        return 'Other'

df['race_label'] = df.apply(label_race, axis=1)

Le DataFrame résultant contiendra le nouveau colonne, race_label, avec les étiquettes de course correspondantes en fonction des critères que vous avez spécifiés.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!