python，最小数组的索引

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])