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
相關文章
相關標籤/搜索