np.argsort()和np.sort()

np.argsort()、np.sort()

1、np.sort()[::-1]

降序排序

 x = np.array([5,4,1,2,3])

In [8]: np.sort(x)[::-1]
Out[8]: array([5, 4, 3, 2, 1])

2、np.sort()[::1]

升序排序

In [7]: np.sort(x)[::1]
Out[7]: array([1, 2, 3, 4, 5])

3、np.argsort()[::-1]

降序排序后返回索引值

In [5]: np.argsort(x)[::-1]
Out[5]: array([0, 1, 4, 3, 2], dtype=int64)

4、np.argsort()[::1]

升序排序后返回索引值

In [6]: np.argsort(x)[::1]
Out[6]: array([2, 3, 4, 1, 0], dtype=int64)

类似的还有np.max(),np.argmax(),np.min(),np.argmin(),在机器学习中经常用到

 

 

 

你可能感兴趣的:(NumPy)