[ x for x in xrange(10) when p(x) ]
gsakkis at rutgers.edu
Thu Nov 10 04:58:16 CET 2005
"bonono at gmail.com" wrote:
> Alex Martelli wrote:
> > This becomes a valid list comprehension by writing 'if' instead of
> > 'when'.
> valid, yes. efficient, I am not sure.
> [ x for x in xrange(10000000) if p(x) ]
> means I need to go through the whole range even if p = lambda x: x < 2
Itertools is your friend in this case:
>>> from itertools import takewhile
>>> list(takewhile(p, xrange(10000000)))
More information about the Python-list