迭代器切片

問題

你想得到一個由迭代器生成的切片對象，但是標準切片操作并不能做到。

解法

函數 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() 會消耗掉傳入的迭代器中的數據。必須考慮到迭代器是不可逆的這個事實。所以如果你需要之后再次訪問這個迭代器的話，那你就得先將它里面的數據放入一個列表中。

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（4.7）：迭代器切片的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。