4.7 迭代器切片

問題

你想得到一個由迭代器生成的切片對象，但是標(biāo)準(zhǔn)切片操作并不能做到。

解決方案

函數(shù) 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
>>>

討論

迭代器和生成器不能使用標(biāo)準(zhǔn)的切片操作，因?yàn)樗鼈兊拈L度事先我們并不知道(并且也沒有實(shí)現(xiàn)索引)。函數(shù) islice() 返回一個可以生成指定元素的迭代器，它通過遍歷并丟棄直到切片開始索引位置的所有元素。然后才開始一個個的返回元素，并直到切片結(jié)束索引位置。

這里要著重強(qiáng)調(diào)的一點(diǎn)是 islice() 會消耗掉傳入的迭代器中的數(shù)據(jù)。必須考慮到迭代器是不可逆的這個事實(shí)。所以如果你需要之后再次訪問這個迭代器的話，那你就得先將它里面的數(shù)據(jù)放入一個列表中。