Pandas: как проверить, равны ли несколько столбцов

Вы можете использовать следующие методы, чтобы проверить, равны ли несколько столбцов в пандах:

Способ 1: проверьте, равны ли все столбцы

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

Способ 2: проверьте, равны ли определенные столбцы

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

В следующих примерах показано, как использовать каждый метод на практике со следующим DataFrame 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

Пример 1. Проверьте, равны ли все столбцы

Мы можем использовать следующий синтаксис, чтобы проверить, одинаково ли значение каждого столбца в DataFrame для каждой строки:

 #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

Если значения каждого столбца равны, соответствующий столбец возвращает True .

В противном случае он возвращает False .

Обратите внимание, что вы можете преобразовать значения True и False в 1 и 0 , используя astype(int) следующим образом:

 #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

Пример 2. Проверьте, равны ли определенные столбцы

Мы можем использовать следующий синтаксис, чтобы проверить, равны ли значения в столбцах A, C и D DataFrame для каждой строки:

 #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

Если значения в столбцах A, C и D равны, соответствующий столбец возвращает True .

В противном случае он возвращает False .

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные задачи в pandas:

Как переименовать столбцы в Pandas
Как добавить столбец в DataFrame Pandas
Как изменить порядок столбцов в Pandas DataFrame