Python: come impostare la variabile locale nella comprensione degli elenchi?

Question

Ho un metodo che prende una lista e restituire un oggetto

# input a list, returns an object 
def map_to_obj(lst): 
    a_list = f(lst) 
    return a_list[0] if a_list else None 

voglio ottenere un elenco che contiene tutti gli elementi mappati che non è None.

Ti piace questa:

v_list = [v1, v2, v3, v4] 

[map_to_obj(v) for v in v_list if map_to_obj(v)] 

Ma sembra non è bene chiamare il metodo map_to_obj due volte nella lista di comprensione.

C'è un modo per avere variabili locali nella comprensione delle liste in modo che possa avere prestazioni migliori?

Oppure il compilatore lo ottimizza automaticamente?

Ecco quello che voglio:

(sml like) 
[let mapped = map_to_obj(v) in for v in v_list if mapped end] 

Answer 1

Usa lista annidata di comprensione:

[x for x in [map_to_obj(v) for v in v_list] if x]

o meglio ancora, una lista di comprensione intorno a un generatore di espressione:

[x for x in (map_to_obj(v) for v in v_list) if x]

Answer 2

È possibile evitare di ri-calcolo utilizzando python built-in filter:

list(filter(lambda t: t is not None, map(map_to_obj, v_list))) 

Answer 3

di lista vanno bene per i casi più semplici, ma a volte un semplice vecchio for Loop è la soluzione più semplice:

other_list = [] 
for v in v_list: 
    obj = map_to_obj(v) 
    if obj: 
     other_list.append(obj) 

Ora, se si vuole veramente un elenco comp e dont vuole costruire un elenco tmp, è possibile utilizzare le versioni iteratore di filter e map:

import itertools as it 
result = list(it.ifilter(None, it.imap(map_to_obj, v_list))) 

o più semplicemente:

import itertools as it 
result = filter(None, it.imap(map_to_obj, v_list))) 

Le versioni iteratore non costruire una lista temporanea, usano valutazione pigra.

Answer 4

ho trovato un modo di usare reduce:

def map_and_append(lst, v): 
    mapped = map_to_obj(v) 
    if mapped is not None: 
     lst.append(mapped) 
    return lst 

reduce(map_and_append, v_list, []) 

Come circa le prestazioni di questo?