我有一個(gè)示例直方圖，其箱如下所示：[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]如果給我一個(gè)像 之類(lèi)的值5.6，我希望能夠獲取給定值將落入直方圖中的 bin 列表的索引，因此在本例中它將i = 4是5。我還可以假設(shè)給定值始終在垃圾箱的范圍內(nèi)。我知道我可以用循環(huán)來(lái)做到這一點(diǎn)并進(jìn)行一些比較檢查，但是有沒(méi)有更快的方法？
2 回答
慕姐8265434
TA貢獻(xiàn)1813條經(jīng)驗(yàn) 獲得超2個(gè)贊
據(jù)我所知，在 Python 中執(zhí)行此操作的最快方法是使用bisect 庫(kù)。它會(huì)給你的值比你想要的值大一，所以你需要減去一。
import bisect
l = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
bisect.bisect(l, 5.6) - 1
# 4
對(duì)于較大的列表，它比 python 循環(huán)快幾個(gè)數(shù)量級(jí)：
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貢獻(xiàn)1830條經(jīng)驗(yàn) 獲得超3個(gè)贊
雖然 Mark 建議使用 bisect，但我還想提到使用 next 而不導(dǎo)入。
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
添加回答
舉報(bào)
0/150
提交
取消