如何获取数组中特定数字的索引？

Question

我想选择数字8的索引而不知道它在数组中的位置。

a = np.arange(10)

Answer 1

您可以像使用np.where一样：

>>> import numpy as np
>>> a = np.array([1,4,8,2,6,7,9,8,7,8,8,9,1,0])
>>> a
array([1, 4, 8, 2, 6, 7, 9, 8, 7, 8, 8, 9, 1, 0])
>>> np.where(a==8)[0]
array([ 2,  7,  9, 10], dtype=int64)