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按行求一个二维数组的最大值方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 探究python中open函数的使用

    探究python中open函数的使用

    这篇文章主要为大家详细介绍了python中open函数的使用方法,对python文件方法open进行深入研究,感兴趣的小伙伴们可以参考一下
    2016-03-03
  • python time.sleep()是睡眠线程还是进程

    python time.sleep()是睡眠线程还是进程

    这篇文章主要介绍了python time.sleep()是睡眠线程还是进程,通过实例代码给大家介绍了Python Sleep休眠函数 ,需要的朋友可以参考下
    2019-07-07
  • Python+Selenium实现浏览器标签页的切换

    Python+Selenium实现浏览器标签页的切换

    在实际工作中,我们经常会遇到页面切换的情况。就比如当点击了某个功能的按钮后,浏览器出现了新的标签页,需要在这些标签页之间进行切换。本文将利用Selenium实现这一功能,需要的可以参考一下
    2022-06-06
  • Pytorch+PyG实现GIN过程示例详解

    Pytorch+PyG实现GIN过程示例详解

    这篇文章主要为大家介绍了Pytorch+PyG实现GIN过程示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-04-04
  • 浅析Python 实现一个自动化翻译和替换的工具

    浅析Python 实现一个自动化翻译和替换的工具

    这篇文章主要介绍了Python 实现一个自动化翻译和替换的工具,非常不错,具有一定的参考借鉴价值,需要的朋友可以参考下
    2019-04-04
  • python实现同一局域网下传输图片

    python实现同一局域网下传输图片

    这篇文章主要为大家详细介绍了python实现同一局域网下传输图片,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2020-03-03
  • matplotlib 画双轴子图无法显示x轴的解决方法

    matplotlib 画双轴子图无法显示x轴的解决方法

    这篇文章主要介绍了matplotlib 画双轴子图无法显示x轴的解决方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-07-07
  • python playwrigh框架入门安装使用

    python playwrigh框架入门安装使用

    这篇文章主要为大家介绍了python playwrigh框架入门的安装使用详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-07-07
  • Pycharm Available Package无法显示/安装包的问题Error Loading Package List解决

    Pycharm Available Package无法显示/安装包的问题Error Loading Package Li

    这篇文章主要介绍了Pycharm Available Package无法显示/安装包的问题Error Loading Package List解决,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-09-09
  • python中必会的四大高级数据类型(字符,元组,列表,字典)

    python中必会的四大高级数据类型(字符,元组,列表,字典)

    这篇文章主要介绍了python中必会的四大高级数据类型(字符,元组,列表,字典),本文通过实例图文相结合给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-05-05

最新评论