Python For Loop list comprehension

Question

I am looking for a possible equivalent of the following loop in python list comprehension.

    for foo in foos:
        if foo.text == expected_text
            return foo
    return []

Something like this.

found_foo = [foo for foo in foos if foo.text == expected_text]

If this possible using list comprehension?

You return either a single foo or an empty list? This is not usually a good design — John La Rooy
– John La Rooy, Commented Aug 2, 2015 at 8:56
I am confused how to skip / terminate the loop when the foo_found is mathced. — A.J.
– A.J., Commented Aug 2, 2015 at 8:58
Yeah found_foo is actually a list with length == 1 and when the match is not found [] with length is returned — A.J.
– A.J., Commented Aug 2, 2015 at 8:59

falsetru · Accepted Answer · 2015-08-02 09:01:59Z

5

return next((foo for foo in foos if foo.text == expected_text), None)

Next will return the first yielded item that meet the condition.

In case of no matched item, next will return the default value None.

answered Aug 2, 2015 at 8:56

falsetru

371k69 gold badges769 silver badges659 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

yes. It avoids running along the whole iterable if the element is found.