python – 从排序列表中获取大于给定数字的第一个元素

发布时间：2020-12-20 12:14:55 所属栏目：Python 来源：网络整理

导读：我有两个清单.列表B就像一个数据库,我需要逐个比较列表A的每个元素.让我们说 B = [0.6,1.7,3,4.5]A = [0.6,0.9,1.2,1.5,2,2.5,4,4.5] B是一个排序列表,因此对于每个A [i],只要算法在B中找到一个 A [i]的数字,就应该将其作为输出返回.所以我的输出应该类似于

我有两个清单.列表B就像一个数据库,我需要逐个比较列表A的每个元素.让我们说

B = [0.6,1.7,3,4.5]
A = [0.6,0.9,1.2,1.5,2,2.5,4,4.5]

B是一个排序列表,因此对于每个A [i],只要算法在B中找到一个>> A [i]的数字,就应该将其作为输出返回.所以我的输出应该类似于：

C = [0.6,4.5,4.5]

能否请您建议最简单的解决方案,尽可能避免嵌套循环？

解决方法

如果您可以使用第三方库,一个解决方案是通过 np.searchsorted NumPy：

import numpy as np

B = np.array([0.6,4.5])
A = [0.6,4.5]

res = B[np.searchsorted(B,A)]

array([ 0.6,3.,4.5])

这将比顺序循环或基于标准库中的bisect的算法更有效.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!