Как найти индекс элемента в массиве с помощью 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),)