快速找到单个值将落入哪个直方图箱的方法？

Question

我有一个示例直方图，其箱如下所示：[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]如果给我一个像 之类的值5.6，我希望能够获取给定值将落入直方图中的 bin 列表的索引，因此在本例中它将i = 4是5。我还可以假设给定值始终在垃圾箱的范围内。我知道我可以用循环来做到这一点并进行一些比较检查，但是有没有更快的方法？

Answer 1

据我所知，在 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)

Answer 2

虽然 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