Эквивалент 'in' для сравнения двух массивов Numpy

Question

В чистом, невекторизованном Python, который я могу использовать,

>>> a = 9
>>> b = [5, 7, 12]
>>> a in b
False

Я бы хотел сделать что-то похожее для массивов в Numpy, т.е.

>>> a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
>>> b = np.array([5, 7, 12])
>>> a in b
np.array([False, False, False, False, True, False, True, False, False, False])

... хотя это не работает.

Есть ли функция или метод, который достигает этого? Если нет, то какой самый простой способ сделать это?

Answer 1

Вы ищете in1d :

>>> import numpy as np
>>> a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
>>> b = np.array([5, 7, 12])
>>> np.in1d( a, b)
array([False, False, False, False,  True, False,  True, False, False, False], dtype=bool)

Answer 2

Вы сравниваете две очень разные вещи. С чистыми списками Python у вас есть int и список. С NumPy, у вас есть два NumPy массивов. Если вы измените a на int, то оно будет работать как положено в numpy.

>>> a = 9
>>> b = np.array([5, 7, 12])
>>> a in b
False

Также обратите внимание, что то, что вы показываете с двумя списками, является довольно интуитивным результатом Возвращенный массив показывает вам, для каждого значения в массиве a, это в b? 5 и 7 есть, остальные нет. Отсюда и данный результат.

Answer 3

Возможно, вы захотите реализовать какой-либо алгоритм поиска строк, если собираетесь проверить, содержит ли одна последовательность другую последовательность. Ссылка из Википедии