如何使用numpy查找数组中元素的索引？

Question

如果设置a=array([1,4,5,57,45,34]) 那么我们如何得到一个元素的索引呢？说5等。

我知道如何使用list.index（ele）获取列表中的索引。 但阵列怎么样？是否有任何类似的功能，.index绝对无法正常工作，因为我试过这个？或者我们必须自己开发功能？

使用perl，C等还有其他类似的问题，但我没有在这个社区中找到使用python的问题。

Answer 1

如果您想要元素5的所有索引，可以使用numpy.where，示例 -

In [1]: from numpy import array

In [2]: a=array([1,4,5,57,45,34])

In [3]: from numpy import where

In [4]: where(a==5)
Out[4]: (array([2], dtype=int64),)