numpy中的argsort()

65 阅读 0 评论 43 点赞

我是靠谱客的博主美好大象，最近开发中收集的这篇文章主要介绍numpy中的argsort()，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

argsort()

argsort函数返回的是数组值从小到大的索引值

Examples：

One dimensional array:一维数组

>>> x = np.array([3, 1, 2])
>>> np.argsort(x)
array([1, 2, 0])

Two-dimensional array:二维数组

>>> x = np.array([[0, 3], [2, 2]])
>>> x
array([[0, 3], [2, 2]])
>>> np.argsort(x, axis=0) #按列排序
array([[0, 1], [1, 0]])
>>> np.argsort(x, axis=1) #按行排序
array([[0, 1], [0, 1]])

Examples：

>>> x = np.array([3, 1, 2])
>>> np.argsort(x) #按升序排列
array([1, 2, 0])
>>> np.argsort(-x) #按降序排列
array([0, 2, 1])
>>> x[np.argsort(x)] #通过索引值排序后的数组
array([1, 2, 3])
>>> x[np.argsort(-x)]
array([3, 2, 1])