np.argmax()函数解析

np.argmax()函数解析

一维

#取出a中元素最大值所对应的索引，此时最大值为6，其对应的位置索引值为4，（索引值默认从0开始）

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

4

二维

#按列比较，输出最大值的索引
a = np.array([[1,2,3],
             [4,1,6]])
b = np.argmax(a,axis=0)
b

array([1, 0, 1])

三维

#先看矩阵的第一行，按列比较，输出最大值的索引；
#然后看矩阵的第二行，按列比较，输出最大值的索引；
#再看矩阵的第三行，按列比较，输出最大值的索引。
a = np.array([
    [
        [1,2,3],
        [2,5,6],
        [4,5,8]
    ],
    [
        [2,7,9],
        [2,5,8],
        [2,4,6]
    ]
             ])
b = np.argmax(a,axis=0)
b

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