Zrozumienie listy w języku Python: funkcja testowa return

Question

Czy istnieje sposób na przetestowanie zwrotu funkcji w zrozumieniu listy (lub dyktowania)? Chciałbym uniknąć napisania tego:

lst = [] 
for x in range(10): 
    bar = foo(x) 
    if bar: 
    lst.append(bar) 

i zamiast tego wykorzystuję zrozumienie list. Oczywiście, nie chcę pisać:

[foo(x) for x in range(10) if foo(x)] 

, więc?

[foo(x) for x in range(10) if ??? ] 

Answer 1

Jak o

filter(None, map(foo, range(10))) 

Jeśli nie chcesz zachować listę pośrednią, wymienić map() z itertools.imap(). I z itertools.ifilter(), całość mogła zostać przekształcona w generator.

itertools.ifilter(None, itertools.imap(foo, range(10))) 

Answer 2

Wystarczy utworzyć generator, aby następnie obliczyć wartości i zbudować filtrowaną listę z generatora.

Przykład:

# create generator via generator expression 
results = (foo(x) for x in xrange(10)) 
# build result list, not including falsy values 
filtered_results = [i for i in results if i] 

Dla porównania:

• https://docs.python.org/3/glossary.html#term-generator
• https://wiki.python.org/moin/Generators
• https://www.python.org/dev/peps/pep-0289/