Advertisement

Python numpy库一种排序并恢复原始顺序的方法(Python 3.x.x版本)

阅读量:

前言

在实际应用中,偶尔会遇到这样的情况:需要先对数据进行排序操作,随后执行特定的处理步骤,最后再将数据还原至初始的排列状态。经过网络搜索后发现,目前尚无现成的示例可供参考,因此在此处进行相关记录。

一维情况分析

复制代码
    import numpy as np
    
    arr = np.array([2, 1, 3, 5, 2, 44, 11])
    print('原来顺序:', arr)
    
    order = np.argsort(-arr)
    arr = sorted(arr, reverse=True)
    print('排序后顺序:',arr)
    
    recovery_arr = np.zeros_like(arr)
    for idx, num in enumerate(arr):
    recovery_arr[order[idx]] = num
    print('回复原来顺序:',recovery_arr)
    
    
      
      
      
      
      
      
      
      
      
      
      
      
      
    

输出结果为:

复制代码
    原来顺序: [ 2  

全部评论 (0)

还没有任何评论哟~