Проверка вхождения в NumPy

Чтобы проверить вхождение элементов массива в набор, используют функцию isin. Она сравнивает ряд с другим списком значений или с отдельным массивом и возвращает логическую маску той же длины: в каждой ячейке True, если значение найдено в наборе, и False в противном случае.

Создадим ряд и набор из массива NumPy, затем выведем маску:

import numpy as np

vals = np.array([1, 2, 3, 4, 5])
needles = np.array([2, 4, 6])
mask = np.isin(vals, needles)
print(mask) # выведет [False  True False  True False]

Второй аргумент может быть обычным списком Python:

import numpy as np

scores = np.array([10, 20, 30, 40])
mask = np.isin(scores, [20, 40])
print(mask) # выведет [False  True False  True]

Для ряда цен [3, 1, 4, 1, 5] и набора [1, 5] выведите логическую маску той же длины.

Создайте ряд [2, 4, 6, 8] и проверьте вхождение в список [4, 8, 10]. Выведите только логический результат.

Задайте ряд букв ['a', 'b', 'c', 'd'] и набор ['b', 'd']. Выведите логическую маску той же длины, что и ряд букв.