我想对一个numpy数组进行排序,并找出每个元素的位置。
我对这个解决方案感到不满意,因为它似乎非常低效。它需要两个独立的步骤来进行排序和搜索。
这种高级索引在存在重复元素的数组中会失败,请看:
numpy.argsort
会告诉我,在排序后的数组中,每个索引对应于未排序数组中的哪个索引。我正在寻找类似于反向操作的东西:对于未排序数组中的每个索引,它在排序后的数组中的位置是什么。a = np.array([1, 4, 2, 3])
# a sorted is [1,2,3,4]
# the 1 goes to index 0
# the 4 goes to index 3
# the 2 goes to index 1
# the 3 goes to index 2
# desired output
[0, 3, 1, 2]
# for comparison, argsort output
[0, 2, 3, 1]
一个简单的解决方案使用numpy.searchsorted
np.searchsorted(np.sort(a), a)
# produces [0, 3, 1, 2]
我对这个解决方案感到不满意,因为它似乎非常低效。它需要两个独立的步骤来进行排序和搜索。
这种高级索引在存在重复元素的数组中会失败,请看:
a = np.array([1, 4, 2, 3, 5])
print(np.argsort(a)[np.argsort(a)])
print(np.searchsorted(np.sort(a),a))
a = np.array([1, 4, 2, 3, 5, 2])
print(np.argsort(a)[np.argsort(a)])
print(np.searchsorted(np.sort(a),a))
argsort
的第二次调用了?通过解释它的工作原理以及为什么它有意义,你可以改进答案 :) - MB-F