numpy里面的argmax函数

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])