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按行求一个二维数组的最大值方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
基于python实现FTP文件上传与下载操作(ftp&sftp协议)
这篇文章主要介绍了基于python实现FTP文件上传与下载操作(ftp&sftp协议),本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-04-04
Python语法糖for else循环语句里的break使用详解
这篇文章主要介绍了Python语法糖之for else循环语句里的break使用详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2023-05-05
Python捕获全局的KeyboardInterrupt异常的方法实现
KeyboardInterrupt异常是Python中的一个标准异常,它通常发生在用户通过键盘中断了一个正在运行的程序,本文主要介绍了Python捕获全局的KeyboardInterrupt异常的方法实现,感兴趣的可以了解一下2024-08-08
python3中关于excel追加写入格式被覆盖问题(实例代码)
这篇文章主要介绍了python3中关于excel追加写入格式被覆盖问题,本文通过实例代码给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下2020-01-01
一篇文章告诉你如何用Python控制Excel实现自动化办公
这篇文章主要介绍了教你怎么用Python处理excel实现自动化办公,文中有非常详细的代码示例,对正在学习python的小伙伴们有非常好的帮助,需要的朋友可以参考下2021-08-08


最新评论