白云岛资源网 Design By www.pvray.com
问题描述:
给定一个二维数组,求每一行的最大值
返回一个列向量
如:
给定数组【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按行求一个二维数组的最大值方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
白云岛资源网 Design By www.pvray.com
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
白云岛资源网 Design By www.pvray.com
暂无评论...
RTX 5090要首发 性能要翻倍!三星展示GDDR7显存
三星在GTC上展示了专为下一代游戏GPU设计的GDDR7内存。
首次推出的GDDR7内存模块密度为16GB,每个模块容量为2GB。其速度预设为32 Gbps(PAM3),但也可以降至28 Gbps,以提高产量和初始阶段的整体性能和成本效益。
据三星表示,GDDR7内存的能效将提高20%,同时工作电压仅为1.1V,低于标准的1.2V。通过采用更新的封装材料和优化的电路设计,使得在高速运行时的发热量降低,GDDR7的热阻比GDDR6降低了70%。