Numpy中找出array中最大值所對應的行和列

Python特別靈活,確定方法不止一種,這裏介紹一種我以爲比較簡單的方法。 以下圖,使用x == np.max(x) 得到一個掩模矩陣,而後使用where方法便可返回最大值對應的行和列。 where返回一個長度爲2的元組,第一個元素保存的是行號,第二個元素保存的是列號。web
相關文章
相關標籤/搜索