限制从python生成器获得的项目数量[重复]

Question

有没有比跟随更简洁的方法来限制生成器的项目数量？

def next_dummy_item():
    for i in range(1, 10):
        yield i

item_count = 0
for item in next_dummy_item(): # can't use slicing here :(
  item_count += 1             
  # process item
  if item_count > 5:
    break

Answer 1

使用itertools.islice:

def next_dummy_item():
    for i in range(1, 10):
        yield i

for item in itertools.islice(next_dummy_item(), 5):
    # process item