Pandas : Comment fusionner des colonnes partageant le même nom

Vous pouvez utiliser la syntaxe de base suivante pour fusionner des colonnes dans un DataFrame pandas qui partagent le même nom de colonne :

#define function to merge columns with same names together
def same_merge(x): return ','.join(x[x.notnull()].astype(str))

#define new DataFrame that merges columns with same names together
df_new = df.groupby(level=0, axis=1).apply(lambda x: x.apply(same_merge, axis=1))

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

Exemple : fusionner des colonnes partageant le même nom dans Pandas

Supposons que nous ayons le DataFrame pandas suivant :

import pandas as pd
import numpy as np

#create DataFrame
df = pd.DataFrame({'A': [5, 6, 8, np.nan, 4, np.nan, np.nan],
                   'A1': [np.nan, 12, np.nan, 10, np.nan, 6, 4],
                   'B': [2, 7, np.nan, np.nan, 2, 4, np.nan],
                   'B1': [5, np.nan, 6, 15, 1, np.nan, 4]})

#rename columns so there are duplicate column names
df.columns = ['A', 'A', 'B', 'B']

#view DataFrame
print(df)

     A     A    B     B
0  5.0   NaN  2.0   5.0
1  6.0  12.0  7.0   NaN
2  8.0   NaN  NaN   6.0
3  NaN  10.0  NaN  15.0
4  4.0   NaN  2.0   1.0
5  NaN   6.0  4.0   NaN
6  NaN   4.0  NaN   4.0

Notez que deux colonnes portent le nom « A » et deux colonnes portent le nom « B ».

Nous pouvons utiliser le code suivant pour fusionner les colonnes qui ont les mêmes noms de colonnes et concaténer leurs valeurs avec une virgule :

#define function to merge columns with same names together
def same_merge(x): return ','.join(x[x.notnull()].astype(str))

#define new DataFrame that merges columns with same names together
df_new = df.groupby(level=0, axis=1).apply(lambda x: x.apply(same_merge, axis=1))

#view new DataFrame
print(df_new)

          A        B
0       5.0  2.0,5.0
1  6.0,12.0      7.0
2       8.0      6.0
3      10.0     15.0
4       4.0  2.0,1.0
5       6.0      4.0
6       4.0      4.0

Le nouveau DataFrame a fusionné les colonnes portant les mêmes noms et concaténé leurs valeurs avec une virgule.

Si vous souhaitez utiliser un séparateur différent, remplacez simplement le séparateur virgule par autre chose dans la fonction same_merge() .

Par exemple, le code suivant montre comment utiliser un séparateur point-virgule à la place :

#define function to merge columns with same names together
def same_merge(x): return ';'.join(x[x.notnull()].astype(str))

#define new DataFrame that merges columns with same names together
df_new = df.groupby(level=0, axis=1).apply(lambda x: x.apply(same_merge, axis=1))

#view new DataFrame
print(df_new)

          A        B
0       5.0  2.0;5.0
1  6.0;12.0      7.0
2       8.0      6.0
3      10.0     15.0
4       4.0  2.0;1.0
5       6.0      4.0
6       4.0      4.0

Le nouveau DataFrame a fusionné les colonnes portant les mêmes noms et concaténé leurs valeurs avec un point-virgule.

Ressources additionnelles

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

Comment supprimer les colonnes en double dans Pandas
Comment répertorier tous les noms de colonnes dans Pandas
Comment trier les colonnes par nom dans Pandas