получить вхождения списка в массиве - python

Question

a = np.array([1,2,4,2,3,4,1])
s = [1,2]

Как я могу получить массив, который сообщает мне, существуют ли элементы в s в a?Вот что я надеюсь получить:

[True, True, False, True, False, False, True]

Answer 1

Использование np.isin

>>> np.isin(a,s)

array([ True,  True, False,  True, False, False,  True])

Answer 2

нет, вы можете сделать это следующим образом

a = [1,2,4,2,3,4,1]
s = [1,2]
t = list(map(lambda a: a in s, a))

, если s больше, set более эффективно

 a = [1,2,4,2,3,4,1]
 s = set([1,2])
 t = list(map(lambda a: a in s, a))

Answer 3

import numpy as np

a = np.array([1, 2, 4, 2, 3, 4, 1])
s = [1, 2]
r = [n in s for n in a]
print(r)  # [True, True, False, True, False, False, True]