排序：返回一個數組，每個元素的新位置

Question

我需要排序數組，同時返回一個包含原始元素排序位置的數組。 （NB不是argsort，所述索引來對數組進行排序）

目前這需要兩個步驟：

1. 一種argsort
2. 一個新的陣列上的散佈操作 即POS [argsort [I] ] =我

我覺得我錯過了這裏的一招。這是一個衆所周知的算法，我忽略了一步就可以實現的算法嗎？

步驟2也可以通過搜索實現，但我認爲分散效率更高。

我已經包含了一些示例python代碼來說明問題。

import numpy as np 

l = [0,-8,1,10,13,2] 

a = np.argsort(l) 
# returns [1 0 2 5 3 4], the order required to sort l 

# init new list to zero 
pos = [0 for x in range(0,len(l))] 

# scatter http://en.wikipedia.org/wiki/Gather-scatter_(vector_addressing) 
for i in range(0,len(l)): 
     pos[a[i]] = i 

print pos 
# prints [1, 0, 2, 4, 5, 3], i.e. each original indexes new position in the sorted array 

尋找對這個問題的引用讓我感到沮喪，也許我錯過了這種類型的操作正確的術語。

任何幫助或指導將不勝感激。

Answer 1

下面是一個簡單的實現，儘管它在任何有意義的意義上都不是「就地」的。我不確定「in-place」是什麼意思，因爲輸出是int類型的np.array，輸入可以包含雙精度。

更新響應@夫的評論和澄清意圖：

#!/usr/bin/env python 

import numpy as np 

unsorted = np.array([0,-8,1,10,13,2]) 

def myargsort(numbers): 
    tuples = enumerate(numbers) # returns iterable of index,value 
    sortedTuples = sorted(tuples,key = lambda pair: pair[1]) 
    sortedNumbers = [num for idx,num in sortedTuples] 
    sortIndexes = [idx for idx,num in sortedTuples] 
    return (sortedNumbers,sortIndexes) 

sortedNums, sortIndices = myargsort(unsorted) 

print(unsorted) 
print(sortedNums) 
print(sortIndices)