python学习笔记（一）argmax()

1，argmax()

（1）针对一维数组：返回数组中最大值元素的索引位置，如下代码所示，输出结果为元素6的索引值：4

import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
print(np.argmax(a)) #输出结果为4

（2）针对二维数组：参数axis可取(0,1)，默认是0，表示数组第几维的最大值。下面分别针对axis=0，axis=1进行举例解释

• axis = 0，简单理解，返回结果为：数组中每一列最大值所在“行”索引值。如下代码所示：第一列最大值是9（所在“行”索引值是1），所以针对该列的输出值为1；第二列最大值为7（所在“行”索引值是2），所以针对该列的输出值为2；同理，针对第三列的输出值为2，针对第四列的输出值为1；所以最终的输出值为(1,2,2,1)

import numpy as np
a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])
print(np.argmax(a, axis=0)) #输出结果为(1,2,2,1)

• axis = 1，简单理解，返回结果为：数组中每一行最大值所在“列”索引值。如下代码所示：第一行最大值为5（所在列索引值为1），所以针对第一行的输出值为1；同理，针对第二，第三行的输出值分别为：0，2

import numpy as np
a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])
print(np.argmax(a, axis=1)) #输出结果为(1,0,2)

（3）针对三维的数组，推荐一篇博文  http://www.cnblogs.com/zhouyang209117/p/6512302.html
 

WX关注博主：挑灯看剑王知事