python，minima数组的索引

Question

我在Jupyter Notebook中使用python 3。

想象一下我有：

import numpy as np

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

我现在想找到数组最小值的索引;在这种情况下，数字1和4.

我尝试使用：

np.argmin(q)

这给出了：

1

不幸的是，np.argmin（）仅在只有一个最小值时才有效。如何找到其他最小值的索引？

Answer 1

使用np.where()：

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

结果：

array([1, 4])