我有一个示例直方图，其箱如下所示：[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]如果给我一个像 之类的值5.6，我希望能够获取给定值将落入直方图中的 bin 列表的索引，因此在本例中它将i = 4是5。我还可以假设给定值始终在垃圾箱的范围内。我知道我可以用循环来做到这一点并进行一些比较检查，但是有没有更快的方法？
2 回答
慕姐8265434
TA贡献1813条经验 获得超2个赞
据我所知，在 Python 中执行此操作的最快方法是使用bisect 库。它会给你的值比你想要的值大一，所以你需要减去一。
import bisect
l = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
bisect.bisect(l, 5.6) - 1
# 4
对于较大的列表，它比 python 循环快几个数量级：
import bisect
l = list(range(100000))
n = 79587.6
%timeit next(x[0]-1 for x in enumerate(l) if x[1] > n)
# 8.33 ms ± 22.8 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
%timeit bisect.bisect(l, n) - 1
# 800 ns ± 0.769 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)
牛魔王的故事
TA贡献1830条经验 获得超3个赞
虽然 Mark 建议使用 bisect，但我还想提到使用 next 而不导入。
value = 5.6
lis = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
next(x[0] - 1 for x in enumerate(lis) if x[1] > value)
# 4
添加回答
举报
0/150
提交
取消