我想在二维 numpy 数组中找到最大值的索引 [i,j]:
a = numpy.array([[1,2,3],[4,3,1]])
我尝试使用 numpy.argsort() 来做到这一点，但它返回一个数组，因为它只能沿着一个轴完成。 一种解决方案是比较 argsort 使用 for 循环返回的所有索引(沿两个轴)的元素，但这似乎有点复杂。也许有一个简单的解决方案？
最佳答案
你想要np.unravel_index . np.argmax将返回一个索引，就像遍历数组的扁平版本一样。 unravel_index 将为您提供 N-D 索引。
a = np.random.randint(0, 10, (4,4))
ind = np.unravel_index(np.argmax(a, axis=None), a.shape) # returns a tuple
关于python - 在矩阵中找到最大值的索引(python)，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/53470744/