返回numpy数组中最大值的索引值
numpy.argmax(a, axis=None, out=None)
返回numpy数组元素位置
numpy.unravel_index(indices, shape, order='C')
numpy获取n维矩阵中的最大值的方法如下:
x = np.array([[2,3,1],
[1,1,0],
[0,4,2]])
# numpy最大值坐标
max_index = np.unravel_index(np.argmax(x, axis=None), x.shape)
# numpy最大值
max_value = x[max_index]
print(max_index, max_value)
输出结果:
(2, 1) 4