Panda: come unire colonne che condividono lo stesso nome

È possibile utilizzare la seguente sintassi di base per unire le colonne in un DataFrame panda che condividono lo stesso nome di colonna:

 #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’esempio seguente mostra come utilizzare questa sintassi nella pratica.

Esempio: unisci colonne che condividono lo stesso nome in Panda

Supponiamo di avere i seguenti panda DataFrame:

 import pandas as pd
import numpy as np

#createDataFrame
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)

     AABB
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

Si noti che due colonne sono denominate “A” e due colonne sono denominate “B”.

Possiamo utilizzare il seguente codice per unire colonne che hanno gli stessi nomi di colonna e concatenare i loro valori con una virgola:

 #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)

          AB
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

Il nuovo DataFrame ha unito le colonne con gli stessi nomi e ha concatenato i loro valori con una virgola.

Se desideri utilizzare un separatore diverso, sostituisci semplicemente il separatore virgola con qualcos’altro nella funzione same_merge() .

Ad esempio, il codice seguente mostra come utilizzare invece un separatore punto e virgola:

 #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)

          AB
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

Il nuovo DataFrame ha unito le colonne con gli stessi nomi e ha concatenato i loro valori con un punto e virgola.

Risorse addizionali

I seguenti tutorial spiegano come eseguire altre operazioni comuni nei panda:

Come rimuovere le colonne duplicate in Pandas
Come elencare tutti i nomi delle colonne in Pandas
Come ordinare le colonne per nome in Pandas