python - 限制从 python 生成器获取的项目数量

Question

这个问题已经有答案了:

How to take the first N items from a generator or list? [duplicate] (8 个回答)

已关闭 9 年前。

是否有比以下更简洁的方法来限制生成器中的项目数量？

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