Création d'une colonne avec des conditions If-Else-Else dans Pandas

Pour créer une nouvelle colonne basée sur une condition if-elif-else , il existe deux approches principales :

Approche non vectorisée

Cette approche consiste à définir une fonction qui opère sur les lignes :

def f(row):
    if row['A'] == row['B']:
        val = 0
    elif row['A'] > row['B']:
        val = 1
    else:
        val = -1
    return val

Ensuite, appliquez-le au dataframe le long des lignes :

df['C'] = df.apply(f, axis=1)

Approche vectorisée

L'approche vectorisée utilise np.where pour créer directement la nouvelle colonne :

df['C'] = np.where(
    df['A'] == df['B'], 0, np.where(
    df['A'] >  df['B'], 1, -1)) 

Cette approche est plus efficace pour les grands ensembles de données.

Voici un exemple utilisant le dataframe fourni :

Cadre de données d'entrée

Frame de données de sortie