获取生成器的子集

import itertools

itertools.islice(mygenerator(), 10)

itertools 有许多用于使用迭代器的实用程序。 islice 使用start、stop和step参数对迭代器进行切片,就像对列表进行切片一样。

澄清上述评论:

from itertools import islice

def fib_gen():
    a, b = 1, 1
    while True:
        yield a
        a, b = b, a + b

assert [1, 1, 2, 3, 5] == list(islice(fib_gen(), 5))