Python数据科学:Numpy库基础

argsort函数

返回的是数组值索引

import numpy as np
x = np.array([3, 1, 4, 2, 5])
# 从小到大
np.argsort(x)
Out[4]: array([1, 3, 0, 2, 4])
# 从大到小
np.argsort(-x)
Out[5]: array([4, 2, 0, 3, 1])