np.argsort

函数解释

np.argsort是NumPy库中的一个函数，用于对数组进行排序并返回排序后的索引。它不会直接对数组进行排序，而是返回一个数组，这个数组中的元素是原数组中元素按升序排序后的索引。

numpy.argsort(a, axis=-1, kind=None, order=None)

参数如下：

• a：要排序的数组
• axis：要排序的轴，默认为 -1，表示最后一个轴
• kind：排序算法的类型，默认为 'quicksort'。可以选择'quicksort'、'mergesort'、'heapsort'和 'stable'
• order：如果数组包含字段，则此参数指定要排序的字段

代码示例

一维数组

import numpy as np

arr = np.array([3, 1, 2])
sorted_indices = np.argsort(arr)
print(sorted_indices)

[1 2 0]

二维数组

import numpy as np

arr = np.array([[3, 1, 2], [6, 4, 5]])
sorted_indices = np.argsort(arr, axis=1)
print(sorted_indices)

[[1 2 0]
 [1 2 0]]

降序排序

import numpy as np

arr = np.array([3, 1, 2])
sorted_indices = np.argsort(arr)[::-1]  ## 或者sorted_indices = np.argsort(-arr)
print(sorted_indices)

[0 2 1]