<分区>
有很多类似的问题,但似乎没有一个对我有帮助。我想返回 numpy 数组中最小值的索引。例如说数组是:
[[4,3,5,1]
[2,6,5,1]
[8,3,2,4]]
我希望我的程序返回 (0,3)。我尝试过使用 argmin,但几乎没有取得任何进展。
请注意,我只想返回一组索引。即上例中不是 (0,3) AND (1,3)。
任何帮助将不胜感激,
jack
<分区>
有很多类似的问题,但似乎没有一个对我有帮助。我想返回 numpy 数组中最小值的索引。例如说数组是:
[[4,3,5,1]
[2,6,5,1]
[8,3,2,4]]
我希望我的程序返回 (0,3)。我尝试过使用 argmin,但几乎没有取得任何进展。
请注意,我只想返回一组索引。即上例中不是 (0,3) AND (1,3)。
任何帮助将不胜感激,
jack
最佳答案
使用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)
关于python - 提取 numpy 数组中最小值的索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40617212/