上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的索引地址,其他元素以此类推。