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

我有一个示例直方图,其箱如下所示:

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

如果给我一个像 之类的值5.6,我希望能够获取给定值将落入直方图中的 bin 列表的索引,因此在本例中它将i = 45。我还可以假设给定值始终在垃圾箱的范围内。我知道我可以用循环来做到这一点并进行一些比较检查,但是有没有更快的方法?


HUWWW
浏览 88回答 2
2回答

慕姐8265434

据我所知,在 Python 中执行此操作的最快方法是使用bisect 库。它会给你的值比你想要的值大一,所以你需要减去一。import bisectl = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]bisect.bisect(l, 5.6) - 1# 4对于较大的列表,它比 python 循环快几个数量级:import bisectl = 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)

牛魔王的故事

虽然 Mark 建议使用 bisect,但我还想提到使用 next 而不导入。value = 5.6lis = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]next(x[0] - 1 for x in enumerate(lis) if x[1] > value)# 4
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python