本篇文章給大家分享的是有關(guān)argmax如何在numpy中使用，小編覺(jué)得挺實(shí)用的，因此分享給大家學(xué)習(xí)，希望大家閱讀完這篇文章后可以有所收獲，話不多說(shuō)，跟著小編一起來(lái)看看吧。

創(chuàng)新互聯(lián)公司專注于網(wǎng)站建設(shè)，為客戶提供網(wǎng)站設(shè)計(jì)、做網(wǎng)站、網(wǎng)頁(yè)設(shè)計(jì)開(kāi)發(fā)服務(wù)，多年建網(wǎng)站服務(wù)經(jīng)驗(yàn)，各類網(wǎng)站都可以開(kāi)發(fā)，品牌網(wǎng)站設(shè)計(jì)，公司官網(wǎng)，公司展示網(wǎng)站，網(wǎng)站設(shè)計(jì)，建網(wǎng)站費(fèi)用，建網(wǎng)站多少錢(qián)，價(jià)格優(yōu)惠，收費(fèi)合理。

從最簡(jiǎn)單的例子出發(fā)

假定現(xiàn)在有一個(gè)數(shù)組a = [3, 1, 2, 4, 6, 1]現(xiàn)在要算數(shù)組a中大數(shù)的索引是多少.這個(gè)問(wèn)題對(duì)于剛學(xué)編程的同學(xué)就能解決.最直接的思路,先假定第0個(gè)數(shù)大,然后拿這個(gè)和后面的數(shù)比,找到大的就更新索引.代碼如下

a = [3, 1, 2, 4, 6, 1]
maxindex = 0
i = 0
for tmp in a:
  if tmp > a[maxindex]:
    maxindex = i
  i += 1
print(maxindex)

這個(gè)問(wèn)題雖然簡(jiǎn)單.但是可以幫助我們理解argmax.

解釋

還是從一維數(shù)組出發(fā).看下面的例子.

import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
print(np.argmax(a))

argmax返回的是大數(shù)的索引.argmax有一個(gè)參數(shù)axis,默認(rèn)是0,表示第幾維的大值.看二維的情況.

import numpy as np
a = np.array([[1, 5, 5, 2],
       [9, 6, 2, 8],
       [3, 7, 9, 1]])
print(np.argmax(a, axis=0))

為了描述方便,a就表示這個(gè)二維數(shù)組.np.argmax(a, axis=0)的含義是a[0][j],a[1][j],a[2][j](j=0,1,2,3)中大值的索引.從a[0][j]開(kāi)始,大值索引最初為(0,0,0,0),拿a[0][j]和a[1][j]作比較,9大于1,6大于5,8大于2,所以大值索引由(0,0,0,0)更新為(1,1,0,1),再和a[2][j]作比較,7大于6,9大于5所以更新為(1,2,2,1).再分析下面的輸出.

import numpy as np
a = np.array([[1, 5, 5, 2],
       [9, 6, 2, 8],
       [3, 7, 9, 1]])
print(np.argmax(a, axis=1))

np.argmax(a, axis=1)的含義是a[i][0],a[i][1],a[i][2],a[i][3](i=0,1,2)中大值的索引.從a[i][0]開(kāi)始,a[i][0]對(duì)應(yīng)的索引為(0,0,0),先假定它就是大值索引(思路和上節(jié)簡(jiǎn)單例子完全一致)拿a[i][0]和a[i][1]作比較,5大于1,7大于3所以大值索引由(0,0,0)更新為(1,0,1),再和a[i][2]作比較,9大于7,更新為(1,0,2),再和a[i][3]作比較,不用更新,最終值為(1,0,2)
再看三維的情況.

import numpy as np
a = np.array([
       [
         [1, 5, 5, 2],
         [9, -6, 2, 8],
         [-3, 7, -9, 1]
       ],

       [
         [-1, 5, -5, 2],
         [9, 6, 2, 8],
         [3, 7, 9, 1]
       ]
      ])
print(np.argmax(a, axis=0))

np.argmax(a, axis=0)的含義是a[0][j][k],a[1][j][k] (j=0,1,2,k=0,1,2,3)中大值的索引.從a[0][j][k]開(kāi)始,a[0][j][k]對(duì)應(yīng)的索引為((0,0,0,0),(0,0,0,0),(0,0,0,0)),拿a[0][j][k]和a[1][j][k]對(duì)應(yīng)項(xiàng)作比較6大于-6,3大于-3,9大于-9,所以更新這幾個(gè)位置的索引,將((0,0,0,0),(0,0,0,0),(0,0,0,0))更新為((0,0,0,0),(0,1,0,0),(1,0,1,0)). 再看axis=1的情況.

import numpy as np
a = np.array([
       [
         [1, 5, 5, 2],
         [9, -6, 2, 8],
         [-3, 7, -9, 1]
       ],

       [
         [-1, 5, -5, 2],
         [9, 6, 2, 8],
         [3, 7, 9, 1]
       ]
      ])
print(np.argmax(a, axis=1))

np.argmax(a, axis=1)的含義是a[i][0][k],a[i][1][k] (i=0,1,k=0,1,2,3)中大值的索引.從a[i][0][k]開(kāi)始,a[i][0][k]對(duì)應(yīng)的索引為((0,0,0,0),(0,0,0,0)),拿a[i][0][k]和a[i][1][k]對(duì)應(yīng)項(xiàng)作比較,9大于1,8大于2,9大于-1,6大于5,2大于-5,8大于2,所以更新這幾個(gè)位置的索引,將((0,0,0,0),(0,0,0,0))更新為((1,0,0,1),(1,1,1,1)),現(xiàn)在大值對(duì)應(yīng)的數(shù)組為((9,5,5,8),(9,6,2,8)).再拿((9,5,5,8),(9,6,2,8))和a[i][2][k]對(duì)應(yīng)項(xiàng)從比較,7大于5,7大于6,9大于2.更新這幾個(gè)位置的索引.將((1,0,0,1),(1,1,1,1))更新為((1,2,0,1),(1,2,2,1)).axis=2的情況也是類似的.

以上就是argmax如何在numpy中使用，小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘９ぷ鲿?huì)見(jiàn)到或用到的。希望你能通過(guò)這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道。

網(wǎng)頁(yè)標(biāo)題：argmax如何在numpy中使用-創(chuàng)新互聯(lián)
本文來(lái)源：http://www.sd-ha.com/article42/iojec.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供品牌網(wǎng)站設(shè)計(jì)、移動(dòng)網(wǎng)站建設(shè)、云服務(wù)器、網(wǎng)站改版、面包屑導(dǎo)航、網(wǎng)站建設(shè)

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來(lái)源： 創(chuàng)新互聯(lián)