Pandas : comment créer une colonne si elle n’existe pas

Vous pouvez utiliser la syntaxe de base suivante pour créer une colonne dans un DataFrame pandas si elle n’existe pas déjà :

df['my_column'] = df.get('my_column', df['col1'] * df['col2']) 

Cette syntaxe particulière crée une nouvelle colonne appelée my_column si elle n’existe pas déjà dans le DataFrame et elle est définie comme le produit des colonnes existantes col1 et col2 .

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Exemple : créer une colonne dans Pandas si elle n’existe pas

Supposons que nous ayons le DataFrame pandas suivant :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'day': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
                   'sales': [4, 6, 5, 8, 14, 13, 13, 12, 9, 8, 19, 14],
                   'price': [1, 2, 2, 1, 2, 4, 4, 3, 3, 2, 2, 3]})

#view DataFrame
print(df)

    day  sales  price
0     1      4      1
1     2      6      2
2     3      5      2
3     4      8      1
4     5     14      2
5     6     13      4
6     7     13      4
7     8     12      3
8     9      9      3
9    10      8      2
10   11     19      2
11   12     14      3

Supposons maintenant que nous essayions d’ajouter une colonne appelée prix si elle n’existe pas déjà et que nous la définissions comme une colonne dans laquelle chaque valeur est égale à 100 :

#attempt to add column called 'price'
df['price'] = df.get('price', 100)    

#view updated DataFrame
print(df)

    day  sales  price
0     1      4      1
1     2      6      2
2     3      5      2
3     4      8      1
4     5     14      2
5     6     13      4
6     7     13      4
7     8     12      3
8     9      9      3
9    10      8      2
10   11     19      2
11   12     14      3

Puisqu’une colonne appelée prix existe déjà, pandas ne l’ajoute tout simplement pas au DataFrame.

Cependant, supposons que nous essayions d’ajouter une nouvelle colonne appelée revenu si elle n’existe pas déjà et que nous la définissions comme une colonne dans laquelle les valeurs sont le produit des colonnes ventes et prix :

#attempt to add column called 'revenue'
df['revenue'] = df.get('revenue', df['sales'] * df['price'])

#view updated DataFrame
print(df)

    day  sales  price  revenue
0     1      4      1        4
1     2      6      2       12
2     3      5      2       10
3     4      8      1        8
4     5     14      2       28
5     6     13      4       52
6     7     13      4       52
7     8     12      3       36
8     9      9      3       27
9    10      8      2       16
10   11     19      2       38
11   12     14      3       42

Cette colonne de revenus est ajoutée au DataFrame car elle n’existait pas déjà.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations courantes dans les pandas :

Comment supprimer des lignes dans Pandas DataFrame en fonction de la condition
Comment filtrer un DataFrame Pandas sur plusieurs conditions
Comment utiliser le filtre « NON IN » dans Pandas DataFrame