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

标签 python python-3.x numpy

假设我们有两个已排序的 numpy 数组,ab

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/

相关文章:

python - IDLE自动挤占大量线路

Python float

python - 如何使用 python 和 beautifulsoup 解析 script 标签

Python3.9 malloc : can't allocate region error 3

python - 从没有缓存选项的命令行调用 Python SimpleHTTPServer

python-3.x - 如何将scrapy数据中的数据发送到django模型?

python - 使用 tweepy 发送 mp4 文件

python-3.x - 从具有不同分隔符的结构化 numpy 数组写入行

python - 将 numpy 数组转换为 CSV 字符串并将 CSV 字符串转换回 numpy 数组

python - 从椭圆生成数组