Comment créer une nouvelle colonne dans Pandas basée sur une logique conditionnelle sur plusieurs colonnes ?-Tutoriel Python-php.cn

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

Barbara Streisand

Libérer： 2024-12-10 19:30:12

original

192 Les gens l'ont consulté

How to Create a New Column in Pandas Based on Conditional Logic Across Multiple Columns?

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)

Copier après la connexion

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!