Метод isin

Метод isin применяется к Index и возвращает булев массив (numpy.ndarray) той же длины, что и исходный индекс. Элементы, которые присутствуют в переданной последовательности значений, получают значение True, остальные - False. Первым параметром передаётся итерируемый объект (список, множество, серия или другой индекс) для проверки.

Синтаксис

index.isin(values)

Пример

Проверим, какие элементы индекса присутствуют в заданном списке:

import pandas as pd

ser = pd.Series([1, 2, 3, 4, 5])
res = ser.index.isin([1, 3, 5])
print(res)

Результат выполнения кода:

[False True False True False]

Пример

Передадим множество для проверки вхождения:

import pandas as pd

ser = pd.Series([10, 20, 30, 40, 50])
res = ser.index.isin({0, 2, 4})
print(res)

Результат выполнения кода:

[ True False True False True]

Пример

Используем другой индекс в качестве проверяемого набора:

import pandas as pd

idx_1 = pd.Index([10, 20, 30, 40, 50])
idx_2 = pd.Index([20, 40, 60])
res = idx_1.isin(idx_2)
print(res)

Результат выполнения кода:

[False True False True False]

Пример

Проверим строковый индекс на наличие определённых меток:

import pandas as pd

ser = pd.Series([1, 2, 3], index=['a', 'b', 'c'])
res = ser.index.isin(['b', 'c', 'd'])
print(res)

Результат выполнения кода:

[False True True]

Пример

Применим результат isin для фильтрации данных:

import pandas as pd

ser = pd.Series([1, 2, 3, 4, 5])
mask = ser.index.isin([0, 2, 4])
filtered = ser[mask]
print(filtered)

Результат выполнения кода:

0	1
2	3
4	5
dtype: int64

Смотрите также

метод unique,
который возвращает уникальные значения индекса
метод duplicated,
который находит дублированные элементы индекса
метод intersection,
который возвращает пересечение двух индексов
метод union,
который объединяет два индекса без дубликатов