python二维数组遍历输出每行最大值_python+numpy按行求一个二维数组的最大值方法...

问题描述:

给定一个二维数组,求每一行的最大值

返回一个列向量

如:

给定数组【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得到的是列向量,而不是行向量,这在其他的函数中也有体现

2)求和以及其他运算可以按照行或者列来,通过指定axis即可

3)通过reshape()来重新返回具体的维度,我们需要的维度。函数的参数可以有一个-1,但只能有一个,表示这个数是未知的

以上这篇python+numpy按行求一个二维数组的最大值方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值