numpy argsort

转自: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得到由小到大的数组(一维不指定轴,高维需要指定轴),排序后的数组中的值在原数组中对应的索引就是此函数的返回值。

 

posted @ 2022-01-18 14:23  lypbendlf  阅读(27)  评论(0编辑  收藏  举报