获取数组中与特定值匹配的元素的索引

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]

但这只是将等于'c'的fac元素重复 -- 没有什么帮助。

有没有使用NumPy实现这个功能的建议？或者我应该对目前的结果感到满意呢？

- Eric Snyder

---

如果您对fac=='c'的True/False值感到满意，请使用np.nonzero(或np.where)获取True值的索引。 - hpaulj

Answer 1

有几种方法可以使用NumPy解决这个问题，具体取决于您的需求，您可以使用以下方式：

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

- MSeifert

Answer 2

你要找的函数是numpy的where函数。

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

np.where(fac=='c')

将返回一个元组，其中包含与该值匹配的索引数组和数据类型。

- Mrl0330