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))
>>> 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對應(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
>>> 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中最大元素對應(yīng)的索引
2.python中找出numpy array數(shù)組的最值及其索引
到此這篇關(guān)于python如何獲得list或numpy數(shù)組中最大元素對應(yīng)的索引的文章就介紹到這了,更多相關(guān)python 獲得list或numpy最大元素索引內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
python錯誤:AttributeError: ''module'' object has no attribute
這篇文章主要介紹了python錯誤:AttributeError: 'module' object has no attribute 'setdefaultencoding'問題的解決方法,需要的朋友可以參考下2014-08-08
pyqt4教程之實現(xiàn)windows窗口小示例分享
這篇文章主要介紹了pyqt4實現(xiàn)windows窗口小示例,需要的朋友可以參考下2014-03-03
Django實現(xiàn)登錄隨機(jī)驗證碼的示例代碼
登錄驗證碼是每個網(wǎng)站登錄時的基本標(biāo)配,這篇文章主要介紹了Django實現(xiàn)登錄隨機(jī)驗證碼的示例代碼,小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2018-06-06
盤點Python加密解密模塊hashlib的7種加密算法(推薦)
這篇文章主要介紹了盤點Python加密解密模塊hashlib的7種加密算法,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2021-04-04
pytorch中DataLoader()過程中遇到的一些問題
這篇文章主要介紹了pytorch中DataLoader()過程中遇到的一些問題,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2021-05-05
Python量化交易實戰(zhàn)之使用Resample函數(shù)轉(zhuǎn)換“日K”數(shù)據(jù)
resample函數(shù)是Python數(shù)據(jù)分析庫Pandas的方法函數(shù),它主要用于轉(zhuǎn)換時間序列的頻次,今天通過本文給大家分享python使用Resample函數(shù)轉(zhuǎn)換時間序列的相關(guān)知識,感興趣的朋友一起看看吧2021-06-06
用PyInstaller把Python代碼打包成單個獨立的exe可執(zhí)行文件
這篇文章主要介紹了用PyInstaller把Python代碼打包成單個獨立的exe可執(zhí)行文件,需要的朋友可以參考下2018-05-05

