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