Esiste un modo per testare il ritorno di una funzione in una lista (o dict) di comprensione? Mi piacerebbe evitare di scrivere che:Comprensione degli elenchi Python: test function return
lst = []
for x in range(10):
bar = foo(x)
if bar:
lst.append(bar)
e utilizzare invece una comprensione di lista. Ovviamente, non voglio scrivere:
[foo(x) for x in range(10) if foo(x)]
così?
[foo(x) for x in range(10) if ??? ]
Come su
filter(None, map(foo, range(10)))
Se non si desidera mantenere la lista intermedia, sostituire map() con itertools.imap(). E con itertools.ifilter(), l'intera cosa potrebbe essere trasformata in un generatore.
Basta creare un generatore per calcolare i valori e creare l'elenco filtrato dal generatore in seguito.
Esempio:
# 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]
Per riferimento:
Perché non si desidera utilizzare il primo modo? È il modo appropriato. –
@Rohit Jain - a seconda di 'foo', potrebbe essere un'operazione costosa che non si desidera eseguire due volte. – mgilson
@mgilson: E nel primo esempio, lo fa solo una volta. –