python如何獲得list或numpy數(shù)組中最大元素對應(yīng)的索引
獲得list中最大元素的索引
aa = [1,2,3,4,5]aa.index(max(aa))
相應(yīng)的最小值使用
aa = [1,2,3,4,5]aa.index(min(aa))
獲得numpy數(shù)組中最大元素的索引
1.可以使用numpy的函數(shù),argmax獲得最大元素的索引,相應(yīng)的獲得最小值的話需要使用argmin。
aa = [1,2,3,4,5]arr_aa = np.array(aa)maxindex = np.argmax(arr_aa )
1.也可以將numpy轉(zhuǎn)為list,然后使用list或者最大值索引的方法獲得最大值。
aa = numpy.array([1,2,3,4,5])
先把a(bǔ)a轉(zhuǎn)換為List,再求索引:
bb = aa.tolist()bb.index(max(bb))
1.python 比較靈活,所以還可以有其他的方法,如使用where函數(shù)。 首先我們可以得到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對應(yīng)行數(shù),后者對應(yīng)列數(shù)
>>> 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中最大元素對應(yīng)的索引 2.python中找出numpy array數(shù)組的最值及其索引
到此這篇關(guān)于python如何獲得list或numpy數(shù)組中最大元素對應(yīng)的索引的文章就介紹到這了,更多相關(guān)python 獲得list或numpy最大元素索引內(nèi)容請搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)!
相關(guān)文章:
