[python] 數組排序之后恢復原來的順序

解決的是這樣一個問題

1.我們對數組\列表進行排序

2.對排序后的數組\列表進行操作

3.我們希望把操作之后的數據恢復原先的順序

 

import numpy as np


values = np.array([0.9,0.1,0.5,0.6])
values_index = np.argsort(values) # 獲取順序
sorted_values = sorted(values)    #排序

post_values = process(values)

output_values = np.zeros_like(values)
for idx, sort in enumerate(values_index): 

 output_values[sort] = post_values[idx] #恢復原始順序