[Numpy-discussion] Find n closest values

[Nicolas P. Rougier]

Hi,

I have an array L with regular spaced values between 0 and width.
I have a (sorted) array I with irregular spaced values between 0 and width.

I would like to find the closest value in I for any value in L.

Currently, I'm using the following script but I wonder if I missed an obvious (and faster) solution:


import numpy as np

def find_closest(A, target):
    idx = A.searchsorted(target)
    idx = np.clip(idx, 1, len(A) - 1)
    left = A[idx - 1]
    right = A[idx]
    idx -= target - left < right - target
    return idx

n, width = 256, 100.0

# 10 random sorted values in [0,width]
I = np.sort(np.random.randint(0,width,10))

# n regular spaced values in [0,width]
L = np.linspace(0, width, n)

print I[find_closest(I,L)]



Nicolas