溫馨提示×

溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務(wù)條款》

python+numpy按行求一個二維數(shù)組的最大值方法

發(fā)布時間：2020-09-22 00:22:19 來源：腳本之家 閱讀：745 作者：月下花弄影 欄目：開發(fā)技術(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)容了，希望能給大家一個參考，也希望大家多多支持億速云。

向AI問一下細節(jié)

免責聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點不代表本網(wǎng)站立場，如果涉及侵權(quán)請聯(lián)系站長郵箱：is@yisu.com進行舉報，并提供相關(guān)證據(jù)，一經(jīng)查實，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手