1、对于一维矩阵:

2、对于二维矩阵:
axis = 0 取出每列最大值的索引

axis = 1 取出每行最大值的索引

3、对于三维矩阵(334):
如下图,axis = 0 时
取出a[1][1]、a[2][1]和a[3][1],组成(3,4)维矩阵,取出每列最大值的索引,结果为:[2 1 0 0]
取出a[1][2]、a[2][2]和a[3][2],组成(3,4)维矩阵,取出每列最大值的索引,结果为:[0 1 2 1]
取出a[1][3]、a[2][3]和a[3][3],组成(3,4)维矩阵,取出每列最大值的索引,结果为:[2 0 2 2]

如下图,axis = 1 时
取出a[1],取出每列最大值的索引,结果为:[1 2 0 1]
取出a[2],取出每列最大值的索引,结果为:[1 0 2 1]
取出a[3],取出每列最大值的索引,结果为:[0 0 2 2]

如下图,a
“死记硬背”式 理解 np.argmax(arr,axis)!!!!!!!!!!!!!!!!!!!!!!!!!
最新推荐文章于 2024-03-22 15:21:36 发布
本文详细解析了np.argmax()函数在不同维度矩阵中的应用,包括一维、二维及三维矩阵的情况。针对axis参数为0、1、2时,如何获取最大值的索引进行详尽阐述。

最低0.47元/天 解锁文章
740

被折叠的 条评论
为什么被折叠?



