La comprensione delle liste pitoniche è possibile con questo ciclo?

Question

Ho una relazione di amore/odio con la comprensione delle liste. Da un lato penso che siano puliti ed eleganti. D'altra parte odio leggerli. (specialmente quelli che non ho scritto) In genere seguo la regola di, rendilo leggibile fino a quando non è richiesta la velocità. Quindi la mia domanda è davvero accademica a questo punto.

Desidero un elenco di stazioni da un tavolo le cui stringhe hanno spesso spazi aggiuntivi. Ho bisogno di quegli spazi spogliati. A volte quelle stazioni sono vuote e non dovrebbero essere incluse.

stations = [] 
for row in data: 
    if row.strip(): 
     stations.append(row.strip()) 

che si traduce in questa lista di comprensione:

stations = [row.strip() for row in data if row.strip()] 

Questo funziona abbastanza bene, ma mi viene in mente che sto facendo striscia due volte. Ho intuito che .strip() non era realmente necessario due volte ed è generalmente più lento dell'assegnazione di una variabile.

stations = [] 
for row in data: 
    blah = row.strip() 
    if blah: 
     stations.append(blah) 

Risulta avevo ragione.

> Striptwice list comp 14.5714301669  
> Striptwice loop 17.9919670399 
> Striponce loop 13.0950567955 

Il tempo visualizzato tra i due segmenti del loop, il 2o (striscia una volta) è più veloce. Nessuna vera sorpresa qui. Sono sorpreso che la comprensione delle liste sia solo leggermente più lenta anche se sta facendo una striscia due volte.

La mia domanda: C'è un modo per scrivere una comprensione di lista che fa la striscia solo una volta?

Risultati:

Ecco i risultati di temporizzazione dei suggerimenti

# @JonClements & @ErikAllik 
> Striptonce list comp 10.7998494348 
# @adhie 
> Mapmethod loop 14.4501044569 

Answer 1

comprensioni nidificate può essere difficile da leggere, quindi la mia prima preferenza sarebbe:

stripped = (x.strip() for x in data) 
stations = [x for x in stripped if x] 

Or , se in linea stripped, si ottiene un singolo (nested) lista di comprensione:

stations = [x for x in (x.strip() for x in data) if x] 

si noti che il primo/interna comprensione è una realtà generatore expressio n, che, in altre parole, è una comprensione di lista pigra; questo per evitare di ripetere due volte.

Answer 2

C'è - creare un generatore delle corde spogliato prima, quindi utilizzare tale:

stations = [row for row in (row.strip() for row in data) if row] 

Si potrebbe anche scrivere i t senza un comp, ad esempio (di swap per imap e rimuovere list per Python 2.x):

stations = list(filter(None, map(str.strip, data))) 

Answer 3

Applicare la striscia a tutti gli elementi utilizzando map() e filtrare dopo.

[item for item in map(lambda x: x.strip(), list) if item]