[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] #恢复原始顺序