函数原型为：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。