学习记录_python之argmax

import numpy as np
x=np.array([[1,2,3],[4,5,9],[9,9,9]])
y1=np.argmax(x)			#表示在所有元素寻找最大值，只返回第一次最大值的位置
y2=np.argmax(x,axis=0)  #axis=0 表示沿着列寻找最大值
y3=np.argmax(x,axis=1)   #axis=1 表示沿着行寻找最大值
print(y1)
print(y2)
print(y3)

输出结果为：
5
[2 2 1]
[2 2 0]

python 中索引值从0开始。