溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

numpy或list數(shù)組中最大元素的索引如何使用python獲取

發(fā)布時間:2020-11-17 14:07:49 來源:億速云 閱讀:788 作者:Leah 欄目:開發(fā)技術

numpy或list數(shù)組中最大元素的索引如何使用python獲???針對這個問題,這篇文章詳細介紹了相對應的分析和解答,希望可以幫助更多想解決這個問題的小伙伴找到更簡單易行的方法。

獲得list中最大元素的索引

aa = [1,2,3,4,5]
aa.index(max(aa)) 

相應的最小值使用

aa = [1,2,3,4,5]
aa.index(min(aa)) 

獲得numpy數(shù)組中最大元素的索引

1.可以使用numpy的函數(shù),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函數(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對應行數(shù),后者對應列數(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)) 

關于numpy或list數(shù)組中最大元素的索引如何使用python獲取問題的解答就分享到這里了,希望以上內容可以對大家有一定的幫助,如果你還有很多疑惑沒有解開,可以關注億速云行業(yè)資訊頻道了解更多相關知識。

向AI問一下細節(jié)

免責聲明:本站發(fā)布的內容(圖片、視頻和文字)以原創(chuàng)、轉載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權內容。

AI