numpy之argmax()函数

语法格式：
numpy.argmax(a,axis)
作用：返回axis轴方向最大值的索引

a ：为所需处理的矩阵
axis ：为处理的轴向，axis=1为横轴方向，方向从左到右；axis=0为纵轴方向，方向从上到下

一维数组：

>>> a=np.array([1,2,3,4,5])
>>> print(np.argmax(a))
4
1
2
二维数组：

>>> b=np.array([[1,0,6,4],[9,8,2,3],[3,1,2,7]])
>>> print(np.argmax(b,axis=0))
[1 1 0 2]
>>> print(np.argmax(b,axis=1))
[2 0 3]
1
2
3
4

https://blog.csdn.net/qq_43636375/article/details/105744725

 

numpy之argmax()函数_花落雨微扬的博客-CSDN博客