np.argmax()函数解析

一维

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

二维

按列比较，输出最大值的索引。

三维

先看矩阵的第一行，按列比较，输出最大值的索引；
然后看矩阵的第二行，按列比较，输出最大值的索引；
再看矩阵的第三行，按列比较，输出最大值的索引。