python+numpy按行求一個(gè)二維數(shù)組的最大值方法
問(wèn)題描述:
給定一個(gè)二維數(shù)組,求每一行的最大值
返回一個(gè)列向量
如:
給定數(shù)組【1,2,3;4,5,3】
返回[3;5]
import numpy as np x = np.array([[1,2,3],[4,5,3]]) # 先求每行最大值得下標(biāo) index_max = np.argmax(x, axis=1)# 其中,axis=1表示按行計(jì)算 print(index_max.shape) max = x[range(x.shape[0]), index_max] print(max) # 注意到這里返回的是行向量 # 這可以是一種通用的方法, # 其中range()可以是一個(gè)列向量,表示0到n # index_max也是一個(gè)列向量,表示具體的坐標(biāo) # 這樣,兩個(gè)坐標(biāo)組合起來(lái)就成為了二維索引 max_ = x[range(x.shape[0]), index_max].reshape(-1,1) print(max_) # 這樣變成了列向量
值得注意的是:
1)np.argmax得到的是列向量,而不是行向量,這在其他的函數(shù)中也有體現(xiàn)
2)求和以及其他運(yùn)算可以按照行或者列來(lái),通過(guò)指定axis即可
3)通過(guò)reshape()來(lái)重新返回具體的維度,我們需要的維度。函數(shù)的參數(shù)可以有一個(gè)-1,但只能有一個(gè),表示這個(gè)數(shù)是未知的
以上這篇python+numpy按行求一個(gè)二維數(shù)組的最大值方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
python time.sleep()是睡眠線(xiàn)程還是進(jìn)程
這篇文章主要介紹了python time.sleep()是睡眠線(xiàn)程還是進(jìn)程,通過(guò)實(shí)例代碼給大家介紹了Python Sleep休眠函數(shù) ,需要的朋友可以參考下2019-07-07Python+Selenium實(shí)現(xiàn)瀏覽器標(biāo)簽頁(yè)的切換
在實(shí)際工作中,我們經(jīng)常會(huì)遇到頁(yè)面切換的情況。就比如當(dāng)點(diǎn)擊了某個(gè)功能的按鈕后,瀏覽器出現(xiàn)了新的標(biāo)簽頁(yè),需要在這些標(biāo)簽頁(yè)之間進(jìn)行切換。本文將利用Selenium實(shí)現(xiàn)這一功能,需要的可以參考一下2022-06-06Pytorch+PyG實(shí)現(xiàn)GIN過(guò)程示例詳解
這篇文章主要為大家介紹了Pytorch+PyG實(shí)現(xiàn)GIN過(guò)程示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-04-04淺析Python 實(shí)現(xiàn)一個(gè)自動(dòng)化翻譯和替換的工具
這篇文章主要介紹了Python 實(shí)現(xiàn)一個(gè)自動(dòng)化翻譯和替換的工具,非常不錯(cuò),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-04-04python實(shí)現(xiàn)同一局域網(wǎng)下傳輸圖片
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)同一局域網(wǎng)下傳輸圖片,文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2020-03-03matplotlib 畫(huà)雙軸子圖無(wú)法顯示x軸的解決方法
這篇文章主要介紹了matplotlib 畫(huà)雙軸子圖無(wú)法顯示x軸的解決方法,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-07-07Pycharm Available Package無(wú)法顯示/安裝包的問(wèn)題Error Loading Package Li
這篇文章主要介紹了Pycharm Available Package無(wú)法顯示/安裝包的問(wèn)題Error Loading Package List解決,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09python中必會(huì)的四大高級(jí)數(shù)據(jù)類(lèi)型(字符,元組,列表,字典)
這篇文章主要介紹了python中必會(huì)的四大高級(jí)數(shù)據(jù)類(lèi)型(字符,元組,列表,字典),本文通過(guò)實(shí)例圖文相結(jié)合給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2021-05-05