python如何获得list或numpy数组中最大元素对应的索引
获得list中最大元素的索引
aa = [1,2,3,4,5] aa.index(max(aa))
相应的最小值使用
aa = [1,2,3,4,5] aa.index(min(aa))
获得numpy数组中最大元素的索引
1.可以使用numpy的函数,argmax获得最大元素的索引,相应的获得最小值的话需要使用argmin。
aa = [1,2,3,4,5] arr_aa = np.array(aa) maxindex = np.argmax(arr_aa )
1.也可以将numpy转为list,然后使用list或者最大值索引的方法获得最大值。
aa = numpy.array([1,2,3,4,5])
先把aa转换为List,再求索引:
bb = aa.tolist() bb.index(max(bb))
1.python 比较灵活,所以还可以有其他的方法,如使用where函数。
首先我们可以得到array在全局和每行每列的最大值(最小值同理)
>>> a = np.arange(9).reshape((3,3)) >>> a array([[0, 1, 2], [3, 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))
参考文章
1.python 寻找list中最大元素对应的索引
2.python中找出numpy array数组的最值及其索引
到此这篇关于python如何获得list或numpy数组中最大元素对应的索引的文章就介绍到这了,更多相关python 获得list或numpy最大元素索引内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
相关文章
PyTorch实现重写/改写Dataset并载入Dataloader
这篇文章主要介绍了PyTorch实现重写/改写Dataset并载入Dataloader,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2020-07-07python+selenium定时爬取丁香园的新型冠状病毒数据并制作出类似的地图(部署到云服务器)
这篇文章主要介绍了python+selenium定时爬取丁香园的新冠病毒每天的数据并制作出类似的地图(部署到云服务器),本文给大家介绍的非常详细,具有一定的参考借鉴价值,需要的朋友可以参考下2020-02-02pandas dataframe rolling移动计算方式
在Pandas中,rolling()方法用于执行移动窗口计算,常用于时间序列数据分析,例如,计算某商品的7天或1个月销售总量,可以通过rolling()轻松实现,该方法的关键参数包括window(窗口大小),min_periods(最小计算周期)2024-09-09django ModelForm修改显示缩略图 imagefield类型的实例
今天小编就为大家分享一篇django ModelForm修改显示缩略图 imagefield类型的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
最新评论