- 深度学习与目标检测:工具、原理与算法
- 涂铭 金智勇
- 200字
- 2021-10-09 14:31:36
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的索引地址,其他元素以此类推。