假设我们有两个已排序的 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 的方法来做到这一点?
最佳答案
你可以使用 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])
关于python - 对于一个排序的numpy数组中的每个元素,如何减去另一个排序数组中的最小较大值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55978595/