Python中有一个很棒的包,名为^{}。
但在我开始讨论之前,解释一下迭代协议是如何在Python中实现的会很有帮助。如果要在容器上提供迭代,可以指定提供iterator type的^{}类方法。"Understanding Python's 'for' statement"是一篇很好的文章,介绍了for-in语句在Python中的实际工作方式,并对迭代器类型的工作方式进行了很好的概述。
请看以下内容:>>> sequence = [1, 2, 3, 4, 5]
>>> iterator = sequence.__iter__()
>>> iterator.next()
1
>>> iterator.next()
2
>>> for number in iterator:
print number
3
4
5
现在回到itertools。该包包含用于各种迭代目的的函数。如果你需要做特殊的排序,这是第一个要研究的地方。
在底部,您可以找到Recipes部分,其中包含使用现有itertools作为构建块创建扩展工具集的方法。
有一个有趣的函数可以满足您的需求:def consume(iterator, n):
'''Advance the iterator n-steps ahead. If n is none, consume entirely.'''
collections.deque(itertools.islice(iterator, n), maxlen=0)
下面是一个快速、可读的示例,说明它是如何工作的:>>> import itertools, collections
>>> def consume(iterator, n):
collections.deque(itertools.islice(iterator, n))
>>> iterator = range(1, 16).__iter__()
>>> for number in iterator:
if (number == 5):
# Disregard 6, 7, 8, 9 (5 doesn't get printed just as well)
consume(iterator, 4)
else:
print number
1
2
3
4
10
11
12
13
14
15