【Python】Numpy中的argmax()函数

描述:在数组里查找相同元素,返回索引
argmax()函数
numpy.argmax(a,axis=None,out=None)
参数:
1.a:我们使用的(输入)数组如果是沿着0轴，则返回每一列最大值的索引,如果是沿着1轴，则返回每一行最大值的索引
2.axis:(可选)

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.out（可选）
如果提供这个参数，那么它的结果将被插入到这个数组中,但是out这个参数的赋值应该与当前数组a相同（维度和数据类型相同）
返回值
返回值是该数组中最大元素的位置，返回值的形式与axis有没有值有关系
注意：
如果一个数组中最大值个数不唯一，那么返回第一个最大值的索引

a=np.array([0,6,9,2,1,9])
print(np.argmax(a))
# 2(返回第一个最大值的索引)