How to achieve python's any() with a custom predicate?
Asked Answered
A

2

29
>>> l = list(range(10))
>>> l
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> if filter(lambda x: x > 10, l):
...     print "foo"
... else:                     # the list will be empty, so bar will be printed
...     print "bar"
... 
bar

I'd like to use any() for this instead, but any() only takes one argument: the iterable. Is there a better way?

Ansate answered 24/7, 2013 at 16:19 Comment(0)
S
52

Use a generator expression as that one argument:

any(x > 10 for x in l)

Here the predicate is in the expression side of the generator expression, but you can use any expression there, including using functions.

Demo:

>>> l = range(10)
>>> any(x > 10 for x in l)
False
>>> l = range(20)
>>> any(x > 10 for x in l)
True

The generator expression will be iterated over until any() finds a True result, and no further:

>>> from itertools import count
>>> endless_counter = count()
>>> any(x > 10 for x in endless_counter)
True
>>> # endless_counter last yielded 11, the first value over 10:
...
>>> next(endless_counter)
12
Stench answered 24/7, 2013 at 16:20 Comment(0)
P
7

Use a generator expression inside of any():

pred = lambda x: x > 10
if any(pred(i) for i in l):
    print "foo"
else:
    print "bar"

This assumes you already have some predicate function you want to use, of course if it is something simple like this you can just use the Boolean expression directly: any(i > 10 for i in l).

Parallelogram answered 24/7, 2013 at 16:20 Comment(0)

© 2022 - 2024 — McMap. All rights reserved.