np.argmax()函数解析

一维

取出a中元素最大值所对应的索引,此时最大值为9,其对应的位置索引值为4,(索引值默认从0开始)
np.argmax()函数解析_第1张图片

二维

按列比较,输出最大值的索引。
np.argmax()函数解析_第2张图片

三维

先看矩阵的第一行,按列比较,输出最大值的索引;
然后看矩阵的第二行,按列比较,输出最大值的索引;
再看矩阵的第三行,按列比较,输出最大值的索引。
np.argmax()函数解析_第3张图片

你可能感兴趣的:(Python)