开发者社区 问答 正文

迭代器切片

你想得到一个由迭代器生成的切片对象,但是标准切片操作并不能做到。

展开
收起
哦哦喔 2020-04-16 21:31:37 1102 分享 版权
1 条回答
写回答
取消 提交回答
  • 函数 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
    >>>
    
    2020-04-16 21:31:51
    赞同 展开评论
问答地址: