Numpy:np.argmax()的用法

函数原型为：numpy.argmax(a, axis=None, out=None)

1.基本用法

x = [[1,3,3],
     [7,5,2]]
print(np.argmax(x))  #3

此时的 x 相当于 [[1, 3, 3], [7, 5, 2]]。对于这个例子，7最大，索引位置为3(这个索引按照递增顺序：0->1, 1->3, 2->3, 3->7, 4->5, 5->2)

2.axis参数

x = [[1,3,3],
     [7,5,2]]
print(np.argmax(x,axis=0)) #[1 1 0]

axis=0表示按列操作，也就是对比当前列，找出最大值的索引。
7>1 返回1，5>3 返回1，2<3 返回0。所以结果为[1 1 0]。

x = [[1,3,4],
     [7,5,2]]
print(np.argmax(x,axis=1)) #[2 0]

axis=1表示按行操作，也就是对比当前行，找出最大值的索引。
第一行：4>3>1 返回4的索引 2，第二行：7>5>2 返回7的索引 0。所以结果为[2 0]。

3. 重复元素

那如果碰到重复最大元素，返回第一个最大值索引即可！

x = np.array([1, 3, 2, 3, 0, 1, 0])
print(x.argmax()) # 1

返回第一个3的索引 1。