python, индексы массива минимумов

Я работаю с Python 3 в Jupyter Notebook.

Представьте, что у меня есть:

import numpy as np

q = np.array([5, 2, 6, 7, 2])

Теперь я хочу найти индексы минимумов массива; в данном случае числа 1 и 4.

Я пытался использовать:

np.argmin(q)

Это дает:

1

К сожалению, np.argmin() работает только тогда, когда есть только один минимум. Как найти индекс другого минимума?


jupyter-notebook python numpy indices minimum
person Marnix    schedule 08.11.2016    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Используйте 1_:

np.where(q == q.min())[0]

Результат :

array([1, 4])
person MMF    schedule 08.11.2016