深度学习与目标检测:工具、原理与算法
上QQ阅读APP看书,第一时间看更新

2.3.9 NumPy中的arg运算

argmax()函数用于求数组中最大值的下标,简单来说,就是求最大的数对应的索引(位置)是多少。


index2 = np.argmax([1,2,6,3,2]) #返回的是2

argmin()函数用于求数组中最小值的下标,用法与argmax()函数类似。


index2 = np.argmin([1,2,6,3,2]) #返回的是0

我们继续探索NumPy矩阵的排序和使用索引。


import numpy as np
x = np.arange(15)
print(x) 
np.random.shuffle(x) #随机打乱
print(x) 
sx = np.argsort(x) #从小到大排序,返回索引值
print(sx)

返回索引值中第一个元素7代表x向量中0的索引地址,第二个元素12代表x向量中1的索引地址,其他元素以此类推。