Python/numpy：獲取數值列表的數組位置

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]),)] 

Answer 2

列表有一個可以使用的index方法。

>>> nameArr = ['josh','is','a','person'] 
>>> # Using map 
>>> map(nameArr.index, ['a', 'is']) 
[2, 1] 
>>> # Using list comprehensions 
>>> [nameArr.index(x) for x in ['a', 'is']] 
[2, 1] 

BTW，index提高ValueError如果元素是不是在列表中。因此，如果您想要將不在列表中的元素提供給索引方法，則可能需要適當地處理該錯誤。

Answer 3

如果數組很大，做的位置很多次，你可以創建值映射到它的指數第一的字典：

d = dict(zip(nameArr, range(len(nameArr)))) 
items = ['a','is'] 
print [d.get(x, None) for x in items] 

Answer 4

東西如：

>>> f_idxs = np.ravel([np.where(master_data==i)[0] for i in search_list])