扫二维码与项目经理沟通
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
小编给大家分享一下python中如何找出numpy array数组的最值及其索引，相信大部分人都还不怎么了解，因此分享这篇文章给大家参考一下，希望大家阅读完这篇文章后大有收获，下面让我们一起去了解一下吧！
在list列表中，max(list)可以得到list的大值，list.index(max(list))可以得到大值对应的索引
但在numpy中的array没有index方法，取而代之的是where，其又是list没有的
首先我们可以得到array在全局和每行每列的大值(最小值同理)
>>> a = np.arange(9).reshape((3,3)) >>> a array([[0, 1, 2], [9, 4, 5], [6, 7, 8]]) >>> print(np.max(a)) #全局大 8 >>> print(np.max(a,axis=0)) #每列大 [6 7 8] >>> print(np.max(a,axis=1)) #每行大 [2 5 8]
然后用where得到大值的索引，返回值中，前面的array对应行数，后者对应列数
>>> print(np.where(a==np.max(a))) (array([2], dtype=int64), array([2], dtype=int64)) >>> print(np.where(a==np.max(a,axis=0))) (array([2, 2, 2], dtype=int64), array([0, 1, 2], dtype=int64))
如果array中有相同的大值，where会将其位置全部给出
>>> a[1,0]=8 >>> a array([[0, 1, 2], [8, 4, 5], [6, 7, 8]]) >>> print(np.where(a==np.max(a))) (array([1, 2], dtype=int64), array([0, 2], dtype=int64))
以上是“python中如何找出numpy array数组的最值及其索引”这篇文章的所有内容，感谢各位的阅读！相信大家都有了一定的了解，希望分享的内容对大家有所帮助，如果还想学习更多知识，欢迎关注创新互联行业资讯频道！
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
微信二维码
移动版官网