[Numpy-discussion] "arg" searchsorted

Charles R Harris charlesr.harris@gmail....
Mon Mar 26 16:57:01 CDT 2012


On Mon, Mar 26, 2012 at 3:34 PM, Bryan Van de Ven <bryanv@continuum.io>wrote:

> 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?
>


An alternate API would be to pass a pair (a, sorter) as the first argument.

Chuck
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://mail.scipy.org/pipermail/numpy-discussion/attachments/20120326/f131a04a/attachment.html 


More information about the NumPy-Discussion mailing list