您只需在序列上使用迭代器,并检查迭代器上的any是否在n次内始终返回True:def check(it, num):
it = iter(it)
return all(any(it) for _ in range(num))
>>> check([1, 1, 0], 2)
True
>>> check([1, 1, 0], 3)
False
这里的关键点是迭代器记住它最后一次调用的位置,这样每次any调用都将从最后一次调用结束的位置开始。并将其包装在all中,确保它在any是{}时尽早退出。在
至少在性能方面,这应该比大多数其他方法更快。但是以可读性为代价。在
如果您希望它比基于map和itertools.repeat的解决方案更快,则:
^{pr2}$
以其他答案为基准:# Second "True" element is in the last place
lst = [1] + [0]*1000 + [1]
%timeit check_map(lst, 2) # 10000 loops, best of 3: 20.3 µs per loop
%timeit check(lst, 2) # 10000 loops, best of 3: 23.5 µs per loop
%timeit many(lst, 2) # 10000 loops, best of 3: 153 µs per loop
%timeit sum(l) >= 2 # 100000 loops, best of 3: 19.6 µs per loop
# Second "True" element is the second item in the iterable
lst = [1, 1] + [0]*1000
%timeit check_map(lst, 2) # 100000 loops, best of 3: 3.05 µs per loop
%timeit check(lst, 2) # 100000 loops, best of 3: 6.39 µs per loop
%timeit many(lst, 2) # 100000 loops, best of 3: 5.02 µs per loop
%timeit sum(lst) >= 2 # 10000 loops, best of 3: 19.5 µs per loop