Pandas: como mesclar colunas que compartilham o mesmo nome

Você pode usar a seguinte sintaxe básica para mesclar colunas em um DataFrame do pandas que compartilha o mesmo nome de coluna:

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

O exemplo a seguir mostra como usar essa sintaxe na prática.

Exemplo: mesclar colunas com o mesmo nome no Pandas

Suponha que temos o seguinte DataFrame do pandas:

 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

Observe que duas colunas são denominadas “A” e duas colunas são denominadas “B”.

Podemos usar o código a seguir para mesclar colunas que possuem os mesmos nomes de colunas e concatenar seus valores com uma vírgula:

 #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

O novo DataFrame mesclava colunas com os mesmos nomes e concatenava seus valores com vírgula.

Se você quiser usar um separador diferente, simplesmente substitua o separador de vírgula por outra coisa na função same_merge() .

Por exemplo, o código a seguir mostra como usar um separador de ponto e vírgula:

 #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

O novo DataFrame mesclou colunas com os mesmos nomes e concatenou seus valores com ponto e vírgula.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Como remover colunas duplicadas no Pandas
Como listar todos os nomes de colunas no Pandas
Como classificar colunas por nome no Pandas