python - 对于一个排序的numpy数组中的每个元素，如何减去另一个排序数组中的最小较大值

Question

假设我们有两个已排序的 numpy 数组，a 和 b。

a = np.array([ 0,  1,  2, 10])
b = np.array([ 7, 13])

对于每个元素数组 a，我想减去 b 中第一个较大的元素以获得如下结果:

>>> f(a, b)
array([-7, -6, -5, -3])

我可以使用低效的 for 循环来做到这一点，但是有没有更 numpythonic 的方法来做到这一点？

Answer 1

你可以使用 searchsorted为了这。它将要求对 b 进行排序并且 a 的值不大于 b 中的最大值。

> a = np.array([0, 1, 2, 10, 12, 5, 7])
> b = np.array([7, 13])
> a - b[np.searchsorted(b, a, side='right')]

array([-7, -6, -5, -3, -1, -2, -6])