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 时<
“死记硬背”式 理解 np.argmax(arr,axis)!!!!!!!!!!!!!!!!!!!!!!!!!
最新推荐文章于 2023-02-28 10:44:06 发布
本文详细解析了np.argmax()函数在不同维度矩阵中的应用,包括一维、二维及三维矩阵的情况。针对axis参数为0、1、2时,如何获取最大值的索引进行详尽阐述。
摘要由CSDN通过智能技术生成