Как получить индексы определенного числа в массиве?

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)