[Numpy-discussion] "arg" searchsorted

Bryan Van de Ven bryanv at continuum.io
Mon Mar 26 17:34:11 EDT 2012


I recently got asked about an "arg" version of searchsorted, basically a 
version that could take a sorter as an additional argument. For instance:

     In [13]: a = np.array([5,6,8,1,6,9,0])

     In [14]: s = np.argsort(a)

     In [17]: s
     Out[17]: array([6, 3, 0, 1, 4, 2, 5])

     In [18]: np.searchsorted(a, [1.5, 7.2], sorter=sorter)
     Out[18]: array([2, 5])

     In [19]: np.searchsorted(a, -1.5, sorter=s)
     Out[19]: 0

     In [20]: np.searchsorted(a, 11.5, sorter=s)
     Out[20]: 7

     In [21]: np.searchsorted(a, 8.5, sorter=s)
     Out[21]: 6

     In [32]: np.searchsorted(a, 6, side='left', sorter=sorter)
     Out[32]: 3

     In [33]: np.searchsorted(a, 6, side='right', sorter=sorter)
     Out[33]: 5

I've already implemented this, I wanted to ping the list and see if 
there were any thoughts about this small feature. Or should I just 
submit a PR for consideration?

Thanks,

Bryan




More information about the NumPy-Discussion mailing list