zl程序教程

您现在的位置是:首页 >  后端

当前栏目

numpy argsort

2023-09-14 09:11:20 时间

转自:https://blog.csdn.net/u011475210/article/details/77770751

1.例子

import numpy as np
x = np.array([3, 1, 2])
y=np.argsort(x)
z=x[y]

>>> y
array([1, 2, 0], dtype=int64)
>>> z
array([1, 2, 3])

argsort过程,第一步先sort得到由小到大的数组(一维不指定轴,高维需要指定轴),排序后的数组中的值在原数组中对应的索引就是此函数的返回值。