Get the first item from an iterable that matches a condition

by

Python 2.6+ and Python 3:

If you want StopIteration to be raised if no matching element is found:

next(x for x in the_iterable if x > 3)

If you want default_value (e.g. None) to be returned instead:

next((x for x in the_iterable if x > 3), default_value)

Note that you need an extra pair of parentheses around the generator expression in this case − they are needed whenever the generator expression isn’t the only argument.

I see most answers resolutely ignore the next built-in and so I assume that for some mysterious reason they’re 100% focused on versions 2.5 and older — without mentioning the Python-version issue (but then I don’t see that mention in the answers that do mention the next built-in, which is why I thought it necessary to provide an answer myself — at least the “correct version” issue gets on record this way;-).

Python <= 2.5

The .next() method of iterators immediately raises StopIteration if the iterator immediately finishes — i.e., for your use case, if no item in the iterable satisfies the condition. If you don’t care (i.e., you know there must be at least one satisfactory item) then just use .next() (best on a genexp, line for the next built-in in Python 2.6 and better).

If you do care, wrapping things in a function as you had first indicated in your Q seems best, and while the function implementation you proposed is just fine, you could alternatively use itertools, a for...: break loop, or a genexp, or a try/except StopIteration as the function’s body, as various answers suggested. There’s not much added value in any of these alternatives so I’d go for the starkly-simple version you first proposed.

More Related Contents:

  1. How to iterate through two lists in parallel?
  2. What exactly are iterator, iterable, and iteration?
  3. Modifying list while iterating [duplicate]
  4. Why can’t I iterate twice over the same data?
  5. Removing from a list while iterating over it
  6. Iterate an iterator by chunks (of n) in Python?
  7. Iterate over all pairs of consecutive items in a list [duplicate]
  8. How can I iterate through two lists in parallel?
  9. Getting number of elements in an iterator in Python
  10. What are iterator, iterable, and iteration?
  11. How to read file N lines at a time?
  12. Update row values where certain condition is met in pandas
  13. How do I iterate through two lists in parallel?
  14. Skip multiple iterations in loop
  15. Is there an expression for an infinite iterator?
  16. When is not a good time to use python generators?
  17. python groupby behaviour?
  18. cleanest way to call one function on a list of items
  19. has_next in Python iterators?
  20. How to iterate across lines in two files simultaneously?
  21. How to read one single line of csv data in Python?
  22. Why can’t I iterate twice over the same iterator? How can I “reset” the iterator or reuse the data?
  23. How can I create a list of elements from an iterator (convert the iterator to a list)?
  24. What is the idiomatic way to iterate over a binary file?
  25. Best way to loop over a python string backwards
  26. Proper way to reset csv.reader for multiple iterations?
  27. One-liner to check whether an iterator yields at least one element?
  28. Confused with python lists: are they or are they not iterators? [duplicate]
  29. What does it mean to “consume” in Python? In an iterator?
  30. hasnext() for Python iterators?

Leave a Comment