Is there a technical reason why searchsorted only works with 1d arrays? In a current problem, I have an array v.shape = (61, 160, 320) and an array a.shape = (1, 160, 320). It would be great if I could do np.searchsorted(a, v, axis=0) to yield a (1, 160, 320) array of the searched indices.