numpy如何获取多维矩阵中最大值的坐标

利用argmax和unravel_index函数便可。任何维度的矩阵均可以。web import numpy as np a = np.array([[1,2,3],[4,5,6],[3,2,1]]) pos = np.unravel_index(np.argmax(a),a.shape) print(pos) --------------- (1, 2) 参考:https://stackover
相关文章
相关标签/搜索