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