我有一個示例直方圖，其箱如下所示：[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
提交
取消