python 对迭代器产生的数据进行切片

我们想对迭代器产生的数据进行切片处理，但是普通的切片操作符在这儿并不管用。

In [1]: def count(n):
   ...:     while True:
   ...:         yield n
   ...:         n += 1
   ...:

In [2]: c = count(0)

In [3]: c[10:20]
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-3-d27b6259daf3> in <module>()
----> 1 c[10:20]

TypeError: 'generator' object is not subscriptable

解决方案：

要对迭代器和生成器做切片操作，itertools.islice()函数是很好的选择。实例如下：

In [5]: import itertools

In [6]: for x in itertools.islice(c, 10, 20):
   ...:     print(x)
   ...:
10
11
12
13
14
15
16
17
18
19

       迭代器和生成器是没法执行普通的切片操作的，这是因为不知道它们的长度具体是多少(而且它们也没有实现索引)，islice产生的结果是一个迭代器，它可以产生出所需要的切片元素，但这是通过访问并丢弃所有起始索引之前的元素来实现的，之后的元素会由islice对象产生出来，直到达到结束索引为止。

       需要强调的是islice会消耗掉所提供的迭代器中的数据。由于迭代器的元素只能访问一次，没法倒回去，如果之后还需要倒回去访问前面的数据，那也许就应该先将数据转到列表中去。