Метод isin
Метод isin применяется к DataFrame
и возвращает таблицу такого же размера с
логическими значениями, где True означает,
что элемент присутствует в переданной
последовательности значений. В качестве
параметра можно передать список, множество,
словарь или Series.
Синтаксис
df.isin(values)
Пример
Давайте проверим, какие элементы таблицы входят в заданный список:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 30, 40, 50]
})
res = df.isin([2, 4, 20])
print(res)
Результат выполнения кода:
a b
0 False False
1 True False
2 False False
3 True False
4 False True
Пример
Можно передать множество для более быстрой проверки:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 30, 40, 50]
})
res = df.isin({2, 4, 10})
print(res)
Результат выполнения кода:
a b
0 False True
1 True False
2 False False
3 True False
4 False False
Пример
С помощью словаря можно задать разные условия для разных столбцов:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 30, 40, 50]
})
res = df.isin({'a': [1, 3], 'b': [20, 50]})
print(res)
Результат выполнения кода:
a b
0 True False
1 False True
2 True False
3 False False
4 False True
Пример
Фильтрация строк по условию с использованием
метода isin:
import pandas as pd
df = pd.DataFrame({
'a': [1, 2, 3, 4, 5],
'b': [10, 20, 30, 40, 50]
})
res = df[df['a'].isin([2, 4])]
print(res)
Результат выполнения кода:
a b
1 2 20
3 4 40
