python使用numpy尋找二維數(shù)組的最值及其下標(biāo)方法分析
一、通過(guò)np.max和np.where尋找【所有滿足條件的解】
通過(guò)np.max()找矩陣的最大值,再通過(guò)np.where獲得最大值的位置,代碼如下:
import numpy as np a = np.random.randint(1, 10, size=12) a = a.reshape((3,4)) print(a) print(np.max(a)) r, c = np.where(a == np.max(a)) print(r,c)
輸出:
[[7 8 9 4]
[9 3 9 3]
[5 6 1 5]]9
[0 1 1] [2 0 2]
輸出的是兩個(gè)array,分別是x和y數(shù)組,即找出了和這個(gè)最值相等的所有位置。
二、通過(guò)np.argmax尋找【第一個(gè)滿足條件的解】
把矩陣展成一維,np.argmax可以返回最大值在這個(gè)一維數(shù)組中第一次出現(xiàn)的位置,用這個(gè)位置除以矩陣的列數(shù),所得的商就是最大值所在的行,所得的余數(shù)就是最大值所在的列。
import numpy as np
a = np.random.randint(1, 10, size=12)
a = a.reshape((3,4))
print(a)
print(np.max(a))
m = np.argmax(a) # 把矩陣?yán)梢痪S,m是在一維數(shù)組中最大值的下標(biāo)
r, c = divmod(m, a.shape[1]) # r和c分別為商和余數(shù),即最大值在矩陣中的行和列
# m是被除數(shù), a.shape[1]是除數(shù)
print(r, c)輸出:
[[5 5 9 7]
[5 5 8 9]
[2 3 9 3]]9
0 2
可以看到只找到了第一個(gè)出現(xiàn)的最大值,后續(xù)的是搜索不到的。
以上就是python使用numpy尋找二維數(shù)組的最值及其下標(biāo)的詳細(xì)內(nèi)容,更多關(guān)于python numpy最值下標(biāo)的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章!
相關(guān)文章
pyqt5 鍵盤監(jiān)聽按下enter 就登陸的實(shí)例
今天小編就為大家分享一篇pyqt5 鍵盤監(jiān)聽按下enter 就登陸的實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-06-06
python神經(jīng)網(wǎng)絡(luò)facenet人臉檢測(cè)及keras實(shí)現(xiàn)
這篇文章主要為大家介紹了python神經(jīng)網(wǎng)絡(luò)facenet人臉檢測(cè)及keras實(shí)現(xiàn),有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-05-05
Python與C語(yǔ)言分別解決完全平方數(shù)問(wèn)題
完全平方數(shù)指用一個(gè)整數(shù)乘以自己例如1*1.2*2.3*3等依次類推,若一個(gè)數(shù)能表示成某個(gè)整數(shù)的平方的形式,則稱這個(gè)數(shù)為完全平方數(shù)。完全平方數(shù)是非負(fù)數(shù),而一個(gè)完全平方數(shù)的項(xiàng)有倆個(gè),讓我們用程序來(lái)練習(xí)一下2022-03-03
手把手教你用Python打造互動(dòng)式中秋節(jié)慶祝小程序
中秋節(jié)將至,本文提供了一個(gè)使用Python開發(fā)的中秋節(jié)慶祝小程序教程,通過(guò)簡(jiǎn)單的步驟,您可以創(chuàng)建一個(gè)具有節(jié)日祝福、互動(dòng)式燈謎游戲和模擬中秋明月動(dòng)態(tài)背景的小程序,文章詳細(xì)介紹了程序的功能、實(shí)現(xiàn)步驟以及如何運(yùn)行程序,需要的朋友可以參考下2024-09-09
Python3 使用map()批量的轉(zhuǎn)換數(shù)據(jù)類型,如str轉(zhuǎn)float的實(shí)現(xiàn)
今天小編就為大家分享一篇Python3 使用map()批量的轉(zhuǎn)換數(shù)據(jù)類型,如str轉(zhuǎn)float的實(shí)現(xiàn),具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-11-11
如何在scrapy中集成selenium爬取網(wǎng)頁(yè)的方法
這篇文章主要介紹了如何在scrapy中集成selenium爬取網(wǎng)頁(yè)的方法,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-11-11

