Python基本函数:【numpy】np.argsort()函数详细解释运算过程

一、np.argsort()用法:

np.argsort(a, axis=-1, kind='quicksort', order=None)

函数功能:将a中的元素从小到大排列,提取其在排列前对应的index(索引)输出。

二、举例:
1、一维数组
先定义一个array数组x,然后使用函数

import numpy as np
x=np.array([1,4,3,-1,6,9])
y=np.argsort(x)
print('一维数组的排序结果:{}'.format(y))

输出结果如下:

一维数组的排序结果:[3 0 2 1 4 5]

中间过程详细解释:
①输入为: [1,4,3,-1,6,9],对应索引为:[0,1,2,3,4,5],即
x[0]=1,x[1]=4,x[2]=3,x[3]=-1,x[4]=6,x[5]=9

②排序结果为: [-1,1,3,4,6,9]
argsort()是提取排序结果元素在排列前对应的index(索引)输出。
根据排序前x[3]=-1可知,排序结果中,元素-1对应排序前的索引3;
根据排序前x[5]=9可知,排序结果中,元素9对应排序前的索引5;
以此类推,输出 排序后元素在排列前对应的index(索引):
即:[3,0,2,1,4,5]

来源:当一颗恒星

物联沃分享整理
物联沃-IOTWORD物联网 » Python基本函数:【numpy】np.argsort()函数详细解释运算过程

发表评论