发布于 2015-08-30 07:37:59 | 157 次阅读 | 评论: 0 | 来源: 网络整理
函数 itertools.islice()
正好适用于在迭代器和生成器上做切片操作。比如:
>>> def count(n):
... while True:
... yield n
... n += 1
...
>>> c = count(0)
>>> c[10:20]
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'generator' object is not subscriptable
>>> # Now using islice()
>>> import itertools
>>> for x in itertools.islice(c, 10, 20):
... print(x)
...
10
11
12
13
14
15
16
17
18
19
>>>
迭代器和生成器不能使用标准的切片操作,因为它们的长度事先我们并不知道(并且也没有实现索引)。
函数 islice()
返回一个可以生成指定元素的迭代器,它通过遍历并丢弃直到切片开始索引位置的所有元素。
然后才开始一个个的返回元素,并直到切片结束索引位置。
这里要着重强调的一点是 islice()
会消耗掉传入的迭代器中的数据。
必须考虑到迭代器是不可逆的这个事实。
所以如果你需要之后再次访问这个迭代器的话,那你就得先将它里面的数据放入一个列表中。