獲取與數值匹配的數組元素的索引

Question

我已經在索引文檔上花了幾天的時間，但沒有找到我要找的內容。

考慮一下：

import numpy 
fac=numpy.asarray(['a','a','a','b','b','c','c','c']) 
f_ind = [x for x in range(len(fac)) if fac[x] == 'c'] 

返回[5,6,7]像我想要的。然而，看起來NumPy數組應該有一種機制以更簡潔（高效的）方式實現相同的事情。布爾數組可能是解決方案的一部分：

ba = (fac == 'c') 
f_vals = fac[ba] 

但是，只有反芻的fac那等於'c'元素 - 不是非常有幫助。

有關如何使用NumPy實現這一點的任何建議？或者我應該對我所擁有的東西感到滿意？

Answer 1

你正在尋找的功能是numpy的地方。

https://docs.scipy.org/doc/numpy-1.12.0/reference/generated/numpy.where.html

np.where(fac=='c') 

將返回含有匹配值，該值的索引的陣列的元組，以及數據類型。

Answer 2

有幾種方式與NumPy的到ADRESS此，根據您的需求，您可以使用：

>>> import numpy as np 
>>> fac = np.asarray(['a','a','a','b','b','c','c','c']) 

• where：

  >>> np.where(fac == 'c') 
  (array([5, 6, 7], dtype=int64),) 
  

• argwhere：

  >>> np.argwhere(fac == 'c') 
  array([[5], 
       [6], 
       [7]], dtype=int64) 
  

• flatnonzero：

  >>> np.flatnonzero(fac == 'c') 
  array([5, 6, 7], dtype=int64)