迭代器切片操作

迭代器对象一般来说是不支持像可迭代对象(list,tuple等)的切片操作。
如下示例:

def count(n):
   while True:
       yield n
       n += 1

c = count(0)
c[10:20]
Trackback(most recent call last):
   File "<stdin>", line 1, in <module>
TypeError: 'generator' object is not subscriptable

itertools模块提供了对迭代器对象的切片操作支持,itertools提供了模块级函数islice。

import itertools
for x in itertools.islice(c, 10, 20):
    print(x)
...
10
11
12
13
14
15
16
17
18
19
posted @ 2019-11-06 21:54  Jeffrey_Yang  阅读(122)  评论(0编辑  收藏  举报