argmax函数笔记-全是细节

Code-Demo:

a = np.array([[6,0], 
			  [2,9]])

print(np.argmax(a))
# 3 （把[[6,0],[2,9]]看成[6,0,2,9]。从0索引，第3个索引的元素最大）

print(np.argmax(a,axis=0))
# [0 1]
# 沿列看，第一列第0个位置（元素'6'）最大，
# 沿列看，第二列第1个位置（元素'9'）最大。依旧是从0开始索引）

print(np.argmax(a,axis=1))
# [0 1]
# 沿行看，第一行第0个位置（元素'6'）最大，
# 沿行看，第二行第1个位置（元素'9'）最大。依旧是从0开始索引）

输出：

3

[0 1]#列

[0 1]#行

参考资料

NumPy学习笔记（四）—— argmax()函数_ac不知深的博客-CSDN博客_argmax

argmax最经典解释_火星种萝卜的博客-CSDN博客_argmax 

详解numpy的argmax的具体使用 - 百度文库