说明 看看数组里面 ,最大的那个元素 的index ,可以指定 axis 例子 以下例子: 1) 不指定 aixs,就会 拉平 数组 2) 指定 0 或者 1 ,就是 分别 第一维, 第二维 3) 指定 -1, -2 就是 倒着来。 更加清晰的例子 >>> a[1][2] = 9 >>> a array([[10, 11, 12], [13, 14, 9]]) >>> np.argmax(a, axis=-1) array([2, 1]) >>>