How to look ahead one element (peek) in a Python generator?

by

For sake of completeness, the more-itertools package (which should probably be part of any Python programmer’s toolbox) includes a peekable wrapper that implements this behavior. As the code example in the documentation shows:

>>> p = peekable(['a', 'b'])
>>> p.peek()
'a'
>>> next(p)
'a'

However, it’s often possible to rewrite code that would use this functionality so that it doesn’t actually need it. For example, your realistic code sample from the question could be written like this:

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

(reader’s note: I’ve preserved the syntax in the example from the question as of when I’m writing this, even though it refers to an outdated version of Python)

More Related Contents:

  1. Understanding generators in Python
  2. Lazy Method for Reading Big File in Python?
  3. Difference between Python’s Generators and Iterators
  4. What’s the purpose of “send” function on Python generators?
  5. How do I know if a generator is empty from the start?
  6. Reading from a frequently updated file
  7. Resetting generator object in Python
  8. How to print a generator expression?
  9. How do I build a numpy array from a generator?
  10. Can iterators be reset in Python?
  11. yield in list comprehensions and generator expressions
  12. Converting “yield from” statement to Python 2.7 code
  13. If range() is a generator in Python 3.3, why can I not call next() on a range?
  14. How to write a generator class?
  15. Length of generator output [duplicate]
  16. Get the nth item of a generator in Python
  17. How to get the n next values of a generator in a list (python)
  18. Why does next raise a ‘StopIteration’, but ‘for’ do a normal return?
  19. Can generators be recursive?
  20. What is the difference between raise StopIteration and a return statement in generators?
  21. Turn functions with a callback into Python generators?
  22. Best way to receive the ‘return’ value from a python generator
  23. Handle an exception thrown in a generator
  24. Handle generator exceptions in its consumer
  25. How to slice a generator object or iterator?
  26. What does the “yield” keyword do in Python?
  27. Why do list comprehensions write to the loop variable, but generators don’t? [duplicate]
  28. Are Generators Threadsafe?
  29. Why can’t generators be pickled?
  30. Python: powerset of a given set with generators [duplicate]

Leave a Comment