On Mon, Mar 26, 2012 at 3:34 PM, Bryan Van de Ven <[email protected]>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
_______________________________________________ NumPy-Discussion mailing list [email protected] http://mail.scipy.org/mailman/listinfo/numpy-discussion
