python+numpy按行求一个二维数组的最大值方法
更新时间:2019年07月09日 10:39:10 作者:月下花弄影
今天小编就为大家分享一篇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按行求一个二维数组的最大值方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
Pycharm Available Package无法显示/安装包的问题Error Loading Package Li
这篇文章主要介绍了Pycharm Available Package无法显示/安装包的问题Error Loading Package List解决,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2020-09-09python中必会的四大高级数据类型(字符,元组,列表,字典)
这篇文章主要介绍了python中必会的四大高级数据类型(字符,元组,列表,字典),本文通过实例图文相结合给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2021-05-05
最新评论