python+numpy按行求一個(gè)二維數(shù)組的最大值方法
問題描述:
給定一個(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)組合起來就成為了二維索引 max_ = x[range(x.shape[0]), index_max].reshape(-1,1) print(max_) # 這樣變成了列向量
值得注意的是:
1)np.argmax得到的是列向量,而不是行向量,這在其他的函數(shù)中也有體現(xiàn)
2)求和以及其他運(yùn)算可以按照行或者列來,通過指定axis即可
3)通過reshape()來重新返回具體的維度,我們需要的維度。函數(shù)的參數(shù)可以有一個(gè)-1,但只能有一個(gè),表示這個(gè)數(shù)是未知的
以上這篇python+numpy按行求一個(gè)二維數(shù)組的最大值方法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
python scrapy腳本報(bào)錯(cuò)問題及解決
這篇文章主要介紹了python scrapy腳本報(bào)錯(cuò)問題及解決,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-02-02
基于python實(shí)現(xiàn)FTP文件上傳與下載操作(ftp&sftp協(xié)議)
這篇文章主要介紹了基于python實(shí)現(xiàn)FTP文件上傳與下載操作(ftp&sftp協(xié)議),本文通過實(shí)例代碼給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-04-04
Python語法糖for?else循環(huán)語句里的break使用詳解
這篇文章主要介紹了Python語法糖之for?else循環(huán)語句里的break使用詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-05-05
Python捕獲全局的KeyboardInterrupt異常的方法實(shí)現(xiàn)
KeyboardInterrupt異常是Python中的一個(gè)標(biāo)準(zhǔn)異常,它通常發(fā)生在用戶通過鍵盤中斷了一個(gè)正在運(yùn)行的程序,本文主要介紹了Python捕獲全局的KeyboardInterrupt異常的方法實(shí)現(xiàn),感興趣的可以了解一下2024-08-08
Python利用手勢識(shí)別實(shí)現(xiàn)貪吃蛇游戲
想必大家都玩過貪吃蛇的游戲吧:通過操縱蛇的移動(dòng)方向能夠讓蛇吃到隨機(jī)出現(xiàn)的食物,吃到的食物越多,蛇就會(huì)變得越長。本文將使用手勢識(shí)別來完成貪吃蛇這個(gè)簡單的游戲,感興趣的可以了解一下2022-04-04
Python用imghdr模塊識(shí)別圖片格式實(shí)例解析
這篇文章主要介紹了Python用imghdr模塊識(shí)別圖片格式實(shí)例解析,具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01
python3中關(guān)于excel追加寫入格式被覆蓋問題(實(shí)例代碼)
這篇文章主要介紹了python3中關(guān)于excel追加寫入格式被覆蓋問題,本文通過實(shí)例代碼給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-01-01
一篇文章告訴你如何用Python控制Excel實(shí)現(xiàn)自動(dòng)化辦公
這篇文章主要介紹了教你怎么用Python處理excel實(shí)現(xiàn)自動(dòng)化辦公,文中有非常詳細(xì)的代碼示例,對(duì)正在學(xué)習(xí)python的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-08-08

