python如何獲得list或numpy數(shù)組中最大元素對(duì)應(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)的獲得最小值的話(huà)需要使用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)換為L(zhǎng)ist,再求索引:
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對(duì)應(yīng)行數(shù),后者對(duì)應(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會(huì)將其位置全部給出
>>> 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中最大元素對(duì)應(yīng)的索引 2.python中找出numpy array數(shù)組的最值及其索引
到此這篇關(guān)于python如何獲得list或numpy數(shù)組中最大元素對(duì)應(yīng)的索引的文章就介紹到這了,更多相關(guān)python 獲得list或numpy最大元素索引內(nèi)容請(qǐng)搜索好吧啦網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持好吧啦網(wǎng)!
相關(guān)文章:
1. jsp實(shí)現(xiàn)textarea中的文字保存換行空格存到數(shù)據(jù)庫(kù)的方法2. ASP實(shí)現(xiàn)加法驗(yàn)證碼3. asp知識(shí)整理筆記4(問(wèn)答模式)4. IntelliJ IDEA 統(tǒng)一設(shè)置編碼為utf-8編碼的實(shí)現(xiàn)5. 解決ajax的delete、put方法接收不到參數(shù)的問(wèn)題方法6. IntelliJ IDEA 2020最新激活碼(親測(cè)有效,可激活至 2089 年)7. jsp EL表達(dá)式詳解8. 詳解idea中web.xml默認(rèn)版本問(wèn)題解決9. idea開(kāi)啟代碼提示功能的方法步驟10. java 優(yōu)雅關(guān)閉線程池的方案
