Numpy学习np.argsort()函数
发布时间
阅读量：
阅读量
在对数组执行排序操作时，有时需要获取排序后的索引信息，例如希望得到数组中数值最小的三个元素的位置，此时可以借助np.argsort()函数来实现。以下将对该函数进行简要说明。
np.argsort(a, axis=-1, kind=‘quicksort’, order=None)
该函数的作用是返回能够使数组有序排列的索引序列。
代码示例：
一维情况：
x = np.random.randint(0,10,10)
print(x)
[9 2 2 7 1 5 1 5 1 2]
y = np.argsort(x)#默认从小到大排列，返回相应索引
print(y)
[4 6 8 1 2 9 5 7 3 0]
print(x[y])
[1 1 1 2 2 2 5 5 7 9]
y = np.argsort(-x)#对x，从大到小排列，返回相应索引
print(y)
[0 3 5 7 1 2 9 4 6 8]
print(x[y])
[9 7 5 5 2 2 2 1 1 1]
全部评论 (0)
还没有任何评论哟~