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 _______________________________________________ NumPy-Discussion mailing list NumPy-Discussion@scipy.org http://mail.scipy.org/mailman/listinfo/numpy-discussion