python使用numpy寻找二维数组的最值及其下标方法分析

 更新时间:2023年08月08日 09:33:54   作者:Eureka  
这篇文章主要为大家介绍了python使用numpy寻找二维数组的最值及其下标实现方法详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪

一、通过np.max和np.where寻找【所有满足条件的解】

通过np.max()找矩阵的最大值,再通过np.where获得最大值的位置,代码如下:

import numpy as np
a = np.random.randint(1, 10, size=12)
a = a.reshape((3,4))
print(a)
print(np.max(a))
r, c = np.where(a == np.max(a))
print(r,c)

输出:

[[7 8 9 4]
 [9 3 9 3]
 [5 6 1 5]]

9

 [0 1 1] [2 0 2]

输出的是两个array,分别是x和y数组,即找出了和这个最值相等的所有位置。

二、通过np.argmax寻找【第一个满足条件的解】

把矩阵展成一维,np.argmax可以返回最大值在这个一维数组中第一次出现的位置,用这个位置除以矩阵的列数,所得的商就是最大值所在的行,所得的余数就是最大值所在的列。

import numpy as np
a = np.random.randint(1, 10, size=12)
a = a.reshape((3,4))
print(a)
print(np.max(a))
m = np.argmax(a)                # 把矩阵拉成一维,m是在一维数组中最大值的下标
r, c = divmod(m, a.shape[1])    # r和c分别为商和余数,即最大值在矩阵中的行和列
                                # m是被除数, a.shape[1]是除数
print(r, c)

输出:

[[5 5 9 7]
 [5 5 8 9]
 [2 3 9 3]]

9

0 2

可以看到只找到了第一个出现的最大值,后续的是搜索不到的。

以上就是python使用numpy寻找二维数组的最值及其下标的详细内容,更多关于python numpy最值下标的资料请关注脚本之家其它相关文章!

相关文章

  • jupyter lab文件导出/下载方式

    jupyter lab文件导出/下载方式

    这篇文章主要介绍了jupyter lab文件导出/下载方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-04-04
  • pytorch使用 to 进行类型转换方式

    pytorch使用 to 进行类型转换方式

    今天小编就为大家分享一篇pytorch使用 to 进行类型转换方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-01-01
  • python3.5+tesseract+adb实现西瓜视频或头脑王者辅助答题

    python3.5+tesseract+adb实现西瓜视频或头脑王者辅助答题

    这篇文章主要介绍了python3.5+tesseract+adb实现西瓜视频或头脑王者辅助答题,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-01-01
  • Pandas读取MySQL数据到DataFrame的方法

    Pandas读取MySQL数据到DataFrame的方法

    今天小编就为大家分享一篇Pandas读取MySQL数据到DataFrame的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-07-07
  • Pandas的DataFrame如何做交集,并集,差集与对称差集

    Pandas的DataFrame如何做交集,并集,差集与对称差集

    这篇文章主要介绍了Pandas的DataFrame如何做交集,并集,差集与对称差集,Python的数据类型集合由不同元素组成的集合,集合中是一组无序排列的可 Hash 的值,可以作为字典的Key,下面来看看文章的详细内容吧
    2022-01-01
  • python使用itchat模块给心爱的人每天发天气预报

    python使用itchat模块给心爱的人每天发天气预报

    这篇文章主要介绍了python使用itchat模块给心爱的人每天发天气预报,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-11-11
  • Python装饰器模式定义与用法分析

    Python装饰器模式定义与用法分析

    这篇文章主要介绍了Python装饰器模式定义与用法,结合实例形式分析了Python装饰器模式的具体定义、使用方法及相关操作技巧,需要的朋友可以参考下
    2018-08-08
  • Python OpenCV对图像进行模糊处理详解流程

    Python OpenCV对图像进行模糊处理详解流程

    OpenCV是一个基于BSD许可(开源)发行的跨平台计算机视觉库,可以运行在Linux、Windows、Android和Mac OS操作系统上。它轻量级而且高效——由一系列 C 函数和少量 C++ 类构成,同时提供了Python、Ruby、MATLAB等语言的接口,实现了图像处理和计算机视觉方面很多通用算法
    2021-10-10
  • Pyqt QImage 与 np array 转换方法

    Pyqt QImage 与 np array 转换方法

    今天小编就为大家分享一篇Pyqt QImage 与 np array 转换方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • 基础的十进制按位运算总结与在Python中的计算示例

    基础的十进制按位运算总结与在Python中的计算示例

    按位运算是计算机学习中的基础,Python完全支持位运算符从而几乎可以直接显示出位运算的结果,这里我们稍微总结一下基础的十进制按位运算总结与在Python中的计算示例
    2016-06-06

最新评论