argmax()函数详解

argmax()函数详解

例子１：

>>> a=np.arange(12).reshape(2,6)
>>> a
array([[ 0,  1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10, 11]])
>>> np.argmax(a,1)
array([5, 5])   #第五个数数值最大

例子２：

>>> a=np.arange(24).reshape(2,3,4)
>>> a
array([[[ 0,  1,  2,  3],
        [ 4,  5,  6,  7],
        [ 8,  9, 10, 11]],

       [[12, 13, 14, 15],
        [16, 17, 18, 19],
        [20, 21, 22, 23]]])
>>> np.argmax(a,1)
array([[2, 2, 2, 2],
       [2, 2, 2, 2]])
>>> np.argmax(a,1).shape
(2, 4)

[[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11]] 中[ 8, 9, 10, 11]是最大的

>>> np.argmax(a,2)
array([[3, 3, 3],      
       [3, 3, 3]])
>>> np.argmax(a,2).shape
(2, 3)

[ 0, 1, 2, 3]中[3]是最大的