python / NumPy : Get array locations of a list of values

Question

假设我有一个值数组 nameArr = ['josh','is','a','person'] 我想要一个像 arrayLocation(nameArr,[ 'a','is']) 返回 [2 1]。

该功能是否已经存在，如果不存在，我如何有效地实现它？

Answer 1

使用 numpy.where

In [17]: nameArr = np.array(['josh','is','a','person'])

In [18]: [np.where(nameArr==i) for i in ['a','is']]
Out[18]: [(array([2]),), (array([1]),)]