Справка по сортировке по номеру

Question

В Numpy, как мне создать массив индексов, которые можно использовать, возвращают значения исходного массива в отсортированном порядке? например:

Source: 
    [[4 2 6 7] 
     [1 4 8 9] 
     [3 1 0 3]] 

Indices: 
    [10 4 9 1 8 11 0 5 2 3 6 7] 

Answer 1

Посмотрите на numpy.argsort - он возвращает индексы, которые будут сортировать ваш массив. Вы также можете указать ось для сортировки. Попробуйте:

a = numpy.asarray([[4, 2, 6, 7], [1, 4, 8, 9], [3, 1, 0, 3]]) 
numpy.argsort(a.flat) 

>> array([10, 4, 9, 1, 8, 11, 0, 5, 2, 3, 6, 7]) 

Answer 2

Ответ находится в руководстве:

src = [[ ... ]] 
ravel_src = np.ravel(src) 
indices = np.argsort(ra)