python中numpy函数：argmax

Nmupy 选取最大值的函数：argmax与amax

argmax和amax两个函数都是与最大值相关的函数，这两这有什么区别呢？
argmax返回的是那个值本身，amax返回的是那个最大值的标签

范例如下：

import numpy as np
a=np.array([[1,2,3],[4,5,6],[7,8,9]])
print(a)
amax_=np.amax(a,axis=1)
print('amax_ : ')
print(amax_)
argmax_=np.argmax(a,axis=1)
print('argmax_ : ')
print(argmax_)

这是我们即将要执行的一段代码，两个函数的参数是相同的，都是将array a输入，axis设为1（在每一行中找最大的，后面还会介绍）

结果是

[[1 2 3]
 [4 5 6]
 [7 8 9]]
amax_ :
[3 6 9]
argmax_ :
[2 2 2]

注意标签是从0开始，所以最大的3，6，9是标签2下面的。

关于axis参数

一个简单的方法：要提取数组a中的某一个数可以用a[x][y][z]的方法。axis的value就是第几个中括号代表的那一串数字。
比如下面的例子：

import numpy as np
a=np.array([[1,2,3],[4,5,6],[7,8,9]])
print(a)
amax_1=np.amax(a,axis=1)
print('amax_ axis=1: ')
print(amax_1)
amax_0=np.amax(a,axis=0)
print('amax_ axis=0: ')
print(amax_0)

得出结果：

[[1 2 3]
 [4 5 6]
 [7 8 9]]
amax_ axis=1:
[3 6 9]
amax_ axis=0:
[7 8 9]