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

Question

我有一个示例直方图，其 bin 如下所示：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

如果给我一个类似 的值5.6，我希望能够获取给定值将落入直方图中的 bin 列表的索引，因此在此示例中它将是i = 4which is 5。我还可以假设给定值将始终在 bin 范围内。我知道我可以通过循环来执行此操作并进行一些比较检查，但是有更快的方法吗？

标签： python

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)