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