• 更新

使用np.unravel_index()函數：

np.unravel_index(A.argmax(), A.shape) # (2, 2)

原理同下

——————————————————————————————
假定矩陣A，數據類型為array

A = [[0. 1. 2. 3.][3. 4. 5. 5.][6. 7. 8. 8.]]

• 返回最大值

h, w = A.shape position = A.argmax() # position = 10 row, col = position // w, position % w # row, col = 2, 2 （不是(2, 3)）

說明：由于A.argmax()返回的是矩陣從上到下，從左到右遍歷的索引值（第一個最大值的索引），因此返回值為一個整數而不是坐標，即便如此，我們也能根據返回的位置結合矩陣形狀，根據取整（//）和取余（%）操作得到目標元素在矩陣中的二維索引值

總結

以上是生活随笔為你收集整理的「Python」超简单！Python返回矩阵最大元素/最小值坐标，三行搞定！（更新：一行搞定！）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。