提取numpy数组中的最小索引

Question

这里有很多类似的问题，但它们似乎都没有帮助我。我想在numpy数组中返回最小值的索引。比如说数组是：

[[4,3,5,1]
 [2,6,5,1]
 [8,3,2,4]]

我希望我的程序返回（0,3）。我已经尝试过使用argmin但是接下来没有取得任何进展。

注意，我只想返回一组索引。 I.E不是上述例子中的（0,3）和（1,3）。

非常感谢任何帮助，

杰克

Answer 1

使用unravel_index：

arr = np.array([[4,3,5,1],[2,6,5,1],[8,3,2,4]])
index = np.unravel_index(arr.argmin(), arr.shape)
# (0, 3)