Найти строки массива NumPy, которые содержат список

Question

У меня есть массив 2D NumPy a и список/набор/1D NumPy массив b. Я хотел бы найти те строки, которые содержат a любой из b, то есть,

import numpy as np 

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

b = np.array([1, 2]) 

# result: [True, False, True] 

Любые намеки?

Answer 1

Вы можете использовать np.in1d, чтобы найти соответствия для любого элемента от b в каждом элементе в a. Теперь np.in1d будет сглаживать массивы, поэтому нам нужно будет изменить форму потом. Наконец, поскольку мы хотим найти соответствие ANY по каждой строке в a, используйте np.any вдоль каждой строки. Таким образом, у нас будет такая реализация:

np.in1d(a,b).reshape(a.shape).any(axis=1)