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))>>> aarray([[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>>> aarray([[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最大元素索引內容請搜索好吧啦網以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持好吧啦網!
相關文章: