Python:获取小于给定数字的最后一个元素的索引

Question

假设我有以下某个测量的时间点的 numpy 数组，可以假设它是从小到大排序的:

time_points = np.array([0.4, 0.8, 1.2, 2.0, 2.4, 2.8, 3.2, 3.6])

我想找索引小于或等于 3 的最后一个元素。在这个例子中，元素是 2.8并且索引是5 .所以我想要一个 python 代码，它给我 5 .一个相对较短的代码可以做到这一点:

index = 0
while time_points[index] <= 3:
    index += 1
index -= 1

但是我的time_points数组可能会变得很长，我想知道是否有更快的方法，就像方法 index() 一样仅在一行中执行此操作？就像是

>>> time_points = np.array([0.4, 0.8, 1.2, 2.0, 2.4, 2.8, 3.2, 3.6])
>>> index = function(time_points, 3)
5

或者多行代码，但使用 numpy库使操作更快。
提前致谢!

Answer 1

numpy.searchsorted从字面上看就是您需要的确切功能。
它使用二分搜索来找到最近的点，它每次将列表“切成”一半，只查看那一半。
它可以这样使用，说 arr 是列表。

>>> arr = np.array([0.4, 0.8, 1.2, 2.0, 2.4, 2.8, 3.2, 3.6])
>>> arr.searchsorted(3, 'right') - 1
5
>>> arr.searchsorted(2.8, 'right') - 1 
5 
>>> arr.searchsorted(0, 'right') - 1 # No greater item
-1

我怀疑是否有更快的 numpy 方法，因为这是一个单行，并使用二进制搜索，这通常是排序数组上最快的搜索方法。