numpy查找2d数组行中的1d数组元素

Question

我想在2d数组的行中查找1d数组的元素。

例

In [1]: import numpy as np 

In [2]: a = np.array([7,7,7]) 

In [3]: a 
Out[3]: array([7, 7, 7]) 

In [4]: b = np.arange(15).reshape(3,5) 

In [5]: b 
Out[5]: 
array([[ 0, 1, 2, 3, 4], 
     [ 5, 6, 7, 8, 9], 
     [10, 11, 12, 13, 14]]) 

行0和的b 2不具备的a相应元素的匹配，但行1具有在位置2的预期输出的匹配：

array([nan, 2, nan]) 

如果给定行中有多个匹配，则应使用第一个匹配的位置。

我可以研究出一个使用普通Python循环的解决方案，但我对矢量化这种方法感兴趣。

Answer 1

比较b和一个明智的元素，然后找到每行的第一个True值索引，或者设置为nan，如果全部为False。

np.where(np.sum(b==a[:,None],1) > 0, np.argmax(b==a[:,None],1), np.nan) 
Out[22]: array([ nan, 2., nan])