如何在Python生成器中展望一个元素(peek)?

我无法弄清楚如何在Python生成器中展望一个元素.一看,它就消失了.

这就是我的意思:

gen = iter([1,2,3])
next_value = gen.next()  # okay, I looked forward and see that next_value = 1
# but now:
list(gen)  # is [2, 3]  -- the first value is gone!

这是一个更实际的例子:

gen = element_generator()
if gen.next_value() == 'STOP':
  quit_application()
else:
  process(gen.next())

任何人都可以帮我写一个发电机,你可以看一个元素前进吗?

解决方法:

Python生成器API是一种方式:你不能推回你读过的元素.但是您可以使用itertools module创建一个新的迭代器并添加元素:

import itertools

gen = iter([1,2,3])
peek = gen.next()
print list(itertools.chain([peek], gen))
上一篇:《剑指Offer》包含min函数的栈(Java 实现)


下一篇:【剑指offer】包含min函数的栈