Numpy搜索排序降序

Question

我试图了解如何正确使用numpy.searchsorted来找到插入元素的正确索引。当我运行以下命令时：

e = np.array([9,8,7,6])
np.searchsorted(e,5)

我得到的输出是0，这是错误的，正如我所期望的，4作为答案。对于按降序排列的数组，搜索排序可用吗？如果需要保持数组按降序排列，我有哪些选项？非常感谢!

Answer 1

您可以搜索反向数组，然后再反转索引：

e = np.array([9,8,7,6])
e.size - np.searchsorted(e[::-1], 5, side = "right")
# 4

注意，这里的side参数需要与直接搜索原始数组相反，如果您希望结果是一致的。

Answer 2

您可以使用sorter可选参数来完成它。根据文档，这通常是np.argsort(e)的结果

array([3, 2, 1, 0])

你可以用这种方式更有效地生产：

np.arange(len(e)-1, -1, -1)

然后：

np.searchsorted(e, 7.5, sorter=np.arange(len(e)-1, -1, -1))

给你两个。