一维数组

返回数组中最大值元素的索引位置

import numpy as np
a = np.array([6, 2, 3, 10, 12, 1])
print(np.argmax(a)) #输出结果为4
# 也可以这样用
print(a.argmax())

二维数组

参数axis可取(0,1)，默认是0，表示数组第几维的最大值。

• axis = 0，简单理解，返回结果为：数组中每一列最大值所在“行”索引值。

  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)
  # 也可以这样用
  print(a.argmax(0))
  

• axis = 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=1)) #输出结果为(1,0,2)
  # 也可以这样用
  print(a.argmax(1))
  

三维数组

可以参考http://www.cnblogs.com/zhouyang209117/p/6512302.html

参考：https://blog.csdn.net/nyist_yangguang/article/details/118521240