python+numpy按行求一個二維數(shù)組的最大值方法
問題描述:
給定一個二維數(shù)組,求每一行的最大值
返回一個列向量
如:
給定數(shù)組【1,2,3;4,5,3】
返回[3;5]
import numpy as np x = np.array([[1,2,3],[4,5,3]]) # 先求每行最大值得下標 index_max = np.argmax(x, axis=1)# 其中,axis=1表示按行計算 print(index_max.shape) max = x[range(x.shape[0]), index_max] print(max) # 注意到這里返回的是行向量 # 這可以是一種通用的方法, # 其中range()可以是一個列向量,表示0到n # index_max也是一個列向量,表示具體的坐標 # 這樣,兩個坐標組合起來就成為了二維索引 max_ = x[range(x.shape[0]), index_max].reshape(-1,1) print(max_) # 這樣變成了列向量
值得注意的是:
1)np.argmax得到的是列向量,而不是行向量,這在其他的函數(shù)中也有體現(xiàn)
2)求和以及其他運算可以按照行或者列來,通過指定axis即可
3)通過reshape()來重新返回具體的維度,我們需要的維度。函數(shù)的參數(shù)可以有一個-1,但只能有一個,表示這個數(shù)是未知的
以上這篇python+numpy按行求一個二維數(shù)組的最大值方法就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
基于python實現(xiàn)FTP文件上傳與下載操作(ftp&sftp協(xié)議)
這篇文章主要介紹了基于python實現(xiàn)FTP文件上傳與下載操作(ftp&sftp協(xié)議),本文通過實例代碼給大家介紹的非常詳細,對大家的學(xué)習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-04-04
Python語法糖for?else循環(huán)語句里的break使用詳解
這篇文章主要介紹了Python語法糖之for?else循環(huán)語句里的break使用詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-05-05
Python捕獲全局的KeyboardInterrupt異常的方法實現(xiàn)
KeyboardInterrupt異常是Python中的一個標準異常,它通常發(fā)生在用戶通過鍵盤中斷了一個正在運行的程序,本文主要介紹了Python捕獲全局的KeyboardInterrupt異常的方法實現(xiàn),感興趣的可以了解一下2024-08-08
python3中關(guān)于excel追加寫入格式被覆蓋問題(實例代碼)
這篇文章主要介紹了python3中關(guān)于excel追加寫入格式被覆蓋問題,本文通過實例代碼給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2020-01-01
一篇文章告訴你如何用Python控制Excel實現(xiàn)自動化辦公
這篇文章主要介紹了教你怎么用Python處理excel實現(xiàn)自動化辦公,文中有非常詳細的代碼示例,對正在學(xué)習python的小伙伴們有非常好的幫助,需要的朋友可以參考下2021-08-08

