1 numpy.sort [от малого к большому]
Возвращает отсортированный массив от меньшего к большему
numpy.sort(a, axis=-1, kind=None, order=None)
axis：Axis along which to sort. If None, the array is flattened before sorting. The default is -1, which sorts along the last axis.
选择排序的轴，如果没有，将数组铺平
a = np.array([[1,4],[3,1]])
# sort along the last axis，对行进行排序，等于axis=1
np.sort(a)
array([[1, 4],[1, 3]])
# 将数组铺平
np.sort(a, axis=None)
array([1, 1, 3, 4])
# sort along the first axis，对列进行排序， axis=0
np.sort(a, axis=0)
array([[1, 1],[3, 4]])
2 numpy.argsort [от меньшего к большему]
Возвращает индекс отсортированного массива от меньшего к большему
numpy.argsort(a, axis=-1, kind=None, order=None)[source]
axis：Axis along which to sort. The default is -1 (the last axis). If None, the flattened array is used.
x = np.array([3, 1, 2])
# 得到索引
np.argsort(x)
array([1, 2, 0])
# 对于二维数组
x = np.array([[0, 3], [2, 2]])
ind = np.argsort(x, axis=0) # sorts along first axis (down)
ind
array([[0, 1],[1, 0]])
# 利用索引得到排序后的结果
np.take_along_axis(x, ind, axis=0)
array([[0, 2], [2, 3]])
3 numpy.take_along_axis
Получить массив по индексу массива
numpy.take_along_axis(arr, indices, axis)
np.argsort()Комбинация и np.take_along_axis()и np.flip()может использоваться для сортировки массива [от меньшего к большему] или [от большого к меньшему].
# 从大到小
index = np.argsort(data)
index_flip = np.flip(index)
data_sort = np.take_along_axis(index_flip)
# 从小到大
index = np.argsort(data)
data_sort = np.take_along_axis(index)
Ссылка на обучение: