[Numpy-discussion] "arg" searchsorted
Charles R Harris
charlesr.harris@gmail....
Mon Mar 26 18:34:36 CDT 2012
On Mon, Mar 26, 2012 at 4:32 PM, Bryan Van de Ven <bryanv@continuum.io>wrote:
> On 3/26/12 4:57 PM, Charles R Harris wrote:
>
>
> 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])
>>
>
> An alternate API would be to pass a pair (a, sorter) as the first argument.
>
> Chuck
>
>
> Sure, that would be easy enough to implement. I don't really have a
> preference, is there a reason you would prefer that API?
>
No, just exploring possibilities. Another would be a different name,
searchargsorted or some such. I actually think that is a better alternative
than the pair, but it would add another method to ndarray.
Chuck
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://mail.scipy.org/pipermail/numpy-discussion/attachments/20120326/093e4ac0/attachment-0001.html
More information about the NumPy-Discussion
mailing list