Как проверить, есть ли запись в DataFrame Pandas?

Question

Как проверить, есть ли запись в панде DataFrame? Например, скажем

import pandas as pd

df = pd.DataFrame({'A' : [1,2,3], 'B' : [4,5,6], 'C' : [7,8,9]})

Как проверить, существует ли запись 1,4 в записи в столбцах A,B, независимо от того, что находится в C?

Answer 1

Вы можете передать словарь ( isin ) со значениями для поиска по столбцу:

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
result = df.isin({'A': [1], 'B': [4]})
print(result)

Вывод

       A      B
0   True   True
1  False  False
2  False  False

После этого вы можете узнать, существует ли запись, используя все :

result = df.isin({'A': [1], 'B': [4]}).all(1)
print(result)

Выходные данные

0     True
1    False
2    False
dtype: bool

Чтобы использовать ее в операторе ifи только в столбцах ['A', 'B'] используйте любой , например:

if df[['A', 'B']].isin({'A': [1], 'B': [4]}).all(1).any():
    print('found')