查找间隔索引的有效方法

Question

我正在用Python编写样条类。 计算样条插值的方法需要最接近的x个数据点的索引。 当前，简化版本如下所示：

def evaluate(x):
    for ii in range(N): # N = len(x_data)
        if x_data[ii] <= x <= x_data[ii+1]:
            return calc(x,ii)

因此，它将遍历x_data点的列表，直到找到x所在的区间的下标ii ，然后在执行样条插值的函数calc使用它。 虽然功能正常，但如果x接近数据集的末尾，这对于大型x_data数组似乎效率不高。 是否有一种更有效或更优雅的方法来执行相同的功能，而不需要迭代检查每个间隔？

注意：可以假定x_data已排序，因此x_data[ii] < x_data[ii+1] ，但不一定相等。

Answer 1

这正是bisect的含义， https： //docs.python.org/2/library/bisect.html

from bisect import bisect
index = bisect(x_data,x)
#I dont think you actually need the value of the 2 closest but if you do here it is
point_less = x_data[index-1]  # note this will break if its index 0 so you probably want a special case for that
point_more = x_data[index]

closest_value = min([point_less,point_more],key=lambda y:abs(x-y))

或者，您应该使用二进制搜索（实际上我很确定那是bisect在幕后使用的）....应该是最坏的情况O(log n) （假设您的输入数组已经排序）