如何在NumPy中对float数组降序排序并获取原位置索引数组？

你可以直接用NumPy的np.argsort()函数实现需求，有两种常用写法：

• 写法一：和你当前的降序排序逻辑匹配，对原数组取负后做升序排序的索引，就是原数组降序的索引：

import numpy as np

# 假设你的原数组是LAM
I = np.argsort(-LAM)
# 验证：用索引得到的排序数组和你原来的结果一致
LAM_Sorted = LAM[I]

• 写法二：先获取升序排序的索引，再反转索引数组得到降序索引：

I = np.argsort(LAM)[::-1]
LAM_Sorted = LAM[I]

两种方式得到的I都和原数组LAM尺寸相同，完全对应MATLAB中[B,I] = sort(A, 'descend')返回的索引向量，满足LAM_Sorted = LAM[I]的关系。

内容的提问来源于stack exchange，提问作者TFAE