numpy裡面的argmax函式

2021-09-30 16:54:54 字數 692 閱讀 5118

numpy裡面的argmax函式

函式原型:

def argmax(a, axis=none, out=none)

a----輸入array

axis----為0代表列方向,為1代表行方向

out----結果寫到這個array裡面

例子:

import numpy as np

a = np.array([[2,4,6,1],[1,5,2,9]])

print a

[[2 4 6 1]

[1 5 2 9]]

np.argmax(a)                  ;;返回array中數值最大數的下標,預設將輸入array視作一維,出現相同的最大,返回第一次出現的

7

np.argmax(a, axis=0)    ;;返回列方向上數值最大值下標,2>1, 4<5, 6>2, 1<9所以結果為0, 1, 0, 1

array([0, 1, 0, 1])

np.argmax(a, axis=1)     ;;返回行方向上數值最大值下標,同理axis = 0情況

array([2, 3])

numpy裡面的argmax函式

函式原型 def argmax a,axis none,out none a 輸入array axis 為0代表列方向,為1代表行方向 out 結果寫到這個array裡面 1 從簡單的一維開始 import numpy as np a np.array 3,1,2,4,6,1 bb np.argma...

關於numpy中的廣播與argmax問題

問題背景 最近正在看faster rcnn的anchor target layer.py原始碼,其中關於numpy的argmax 問題搞得很混亂。此問題理清楚還需要高清楚numpy的索引的運作原理。a np.array 1,3,5 2,2,9 a.argmax axis 0 array 1,0,1 ...

列舉裡面的介面的實現

列舉式乙個類,這個類,同樣也有類改由的功能,今天來說說列舉來實現介面。下面還是來看看 吧.首先我們定義乙個介面 介面裡面由兩個抽象方法.如下.public inte ce enuminte ce下滿我們來定義我們的列舉型別的類,分別由三個物件,分別有自己的構造方法.如下.public enum de...