Pandas: como verificar se várias colunas são iguais

Você pode usar os seguintes métodos para verificar se várias colunas são iguais em pandas:

Método 1: verifique se todas as colunas são iguais

 df[' matching '] = df. eq (df. iloc [:, 0], axis= 0 ). all (1)

Método 2: verifique se colunas específicas são iguais

 df[' matching '] = df. apply ( lambda x: x.col1 == x.col3 == x.col4 , axis= 1 )

Os exemplos a seguir mostram como usar cada método na prática com o seguinte DataFrame do pandas:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' A ': [4, 0, 3, 3, 6, 8, 7],
                   ' B ': [4, 2, 3, 5, 6, 4, 7],
                   ' C ': [4, 0, 3, 3, 5, 10, 7],
                   ' D ': [4, 0, 3, 3, 3, 8, 7]})

#view DataFrame
print (df)

   ABCD
0 4 4 4 4
1 0 2 0 0
2 3 3 3 3
3 3 5 3 3
4 6 6 5 3
5 8 4 10 8
6 7 7 7 7

Exemplo 1: Verifique se todas as colunas são iguais

Podemos usar a seguinte sintaxe para verificar se o valor de cada coluna no DataFrame é igual para cada linha:

 #create new column that checks if all columns match in each row
df[' matching '] = df. eq (df. iloc [:, 0], axis= 0 ). all (1)

#view updated DataFrame
print (df)

   ABCD matching
0 4 4 4 4 True
1 0 2 0 0 False
2 3 3 3 3 True
3 3 5 3 3 False
4 6 6 5 3 False
5 8 4 10 8 False
6 7 7 7 7 True

Se o valor de cada coluna for igual, a coluna correspondente retornará True .

Caso contrário, retorna False .

Observe que você pode converter valores True e False em 1 e 0 usando astype(int) da seguinte forma:

 #create new column that checks if all columns match in each row
df[' matching '] = df. eq (df. iloc [:, 0], axis= 0 ). all (1). astype (int)

#view updated DataFrame
print (df)

   ABCD matching
0 4 4 4 4 1
1 0 2 0 0 0
2 3 3 3 3 1
3 3 5 3 3 0
4 6 6 5 3 0
5 8 4 10 8 0
6 7 7 7 7 1

Exemplo 2: Verifique se colunas específicas são iguais

Podemos usar a seguinte sintaxe para verificar se os valores nas colunas A, C e D do DataFrame são iguais para cada linha:

 #create new column that checks if values in columns A, C, and D are equal
df[' matching '] = df. apply ( lambda x: x. A == x. C == x. D , axis= 1 )

#view updated DataFrame
print (df)

   ABCD matching
0 4 4 4 4 True
1 0 2 0 0 True
2 3 3 3 3 True
3 3 5 3 3 True
4 6 6 5 3 False
5 8 4 10 8 False
6 7 7 7 7 True

Se os valores nas colunas A, C e D forem iguais, a coluna correspondente retornará True .

Caso contrário, retorna False .

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras tarefas comuns em pandas:

Como renomear colunas no Pandas
Como adicionar uma coluna a um DataFrame do Pandas
Como alterar a ordem das colunas no Pandas DataFrame