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]

你可能感兴趣的:(numpy函数,python,numpy)