一尘不染

在Python中,如何在排序列表中找到第一个值的索引大于阈值?

algorithm

在Python中,如何在排序列表中找到第一个值的索引大于阈值?

我可以想到几种实现此目的的方法(线性搜索,手写二分法,..),但是我正在寻找一种干净且合理有效的方法。由于这可能是一个非常普遍的问题,所以我相信经验丰富的SOers可以为您提供帮助!

谢谢!


阅读 876

收藏
2020-07-28

共1个答案

一尘不染

看一看开张

import bisect

l = [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

bisect.bisect(l, 55) # returns 7

与线性搜索进行比较:

timeit bisect.bisect(l, 55)
# 375ns


timeit next((i for i,n in enumerate(l) if n > 55), len(l))
# 2.24us


timeit next((l.index(n) for n in l if n > 55), len(l))
# 1.93us
2020-07-28