python - 控制生成器的迭代范围？

Question

有没有办法可以控制Python生成器对象的迭代范围？

示例:假设一个生成器由 2000 个项目组成，我希望一个循环从 1 迭代到 100，另一个循环从 101 迭代到 200，依此类推。

一个简单的for生成器中的项目循环只会耗尽生成器

Answer 1

itertools.islice(generator, start, stop, step) 是您想要使用的。

例如:

def generator():
  n = 1
  while n:
    yield n
    n += 1

items = generator()
for item in itertools.islice(items, 100):
  print ('first', item)
for item in itertools.islice(items, 100):
  print ('second', item)

编辑:上面的当然只能让你每次拿一百。如果您知道您的生成器是有限的，我会将其放入一个列表中，然后迭代它的范围。然后你可以有任意和重叠的范围。