Come posso gestire le eccezioni in una list comprehension in Python?

Question

Ho una comprensione di lista in Python in cui ogni iterazione può generare un'eccezione.

Per esempio, se ho:

eggs = (1,3,0,3,2) 

[1/egg for egg in eggs] 

Prendo un'eccezione ZeroDivisionError nel 3 ° elemento.

Come posso gestire questa eccezione e continuare l'esecuzione della comprensione della lista?

L'unico modo che posso pensare è quello di utilizzare una funzione di supporto:

def spam(egg): 
    try: 
     return 1/egg 
    except ZeroDivisionError: 
     # handle division by zero error 
     # leave empty for now 
     pass 

Ma questo sembra un po 'ingombrante per me.

C'è un modo migliore per farlo in Python?

Nota: Questo è un semplice esempio (vedi "per esempio" di cui sopra), che riuscii perché il mio esempio reale richiede un po 'di contesto. Non mi interessa evitare gli errori di divisione per zero, ma nel gestire le eccezioni nella comprensione di una lista.

Answer 1

Non c'è espressione incorporata in Python che ti permette di ignorare un'eccezione (o restituire valori alternativi & c in caso di eccezioni), quindi è impossibile, letteralmente parlando, "gestire le eccezioni in una comprensione di lista" perché una comprensione di lista è un'espressione che contiene altre espressione, niente di più (cioè, no dichiarazioni, e solo le istruzioni possono catturare/ignorare/gestire eccezioni).

Le chiamate di funzione sono espressione e i corpi di funzione possono includere tutte le istruzioni desiderate, quindi delegare la valutazione della sotto-espressione soggetta a eccezioni a una funzione, come hai notato, è una soluzione possibile (altri, quando possibile, sono i controlli sui valori che potrebbero provocare eccezioni, come suggerito anche in altre risposte).

Le risposte corrette alla domanda "come gestire le eccezioni in una lista di comprensione" stanno tutte esprimendo parte di tutta questa verità: 1) letteralmente, cioè lessicamente nella comprensione stessa, non è possibile; 2) in pratica, deleghi il lavoro a una funzione o controlli i valori soggetti a errori quando ciò è fattibile. La tua ripetuta affermazione che questa non è una risposta è quindi infondata.

Answer 2

È possibile utilizzare

[1/egg for egg in eggs if egg != 0] 

questo sarà semplicemente saltare elementi che sono pari a zero.

Answer 3

No, non c'è un modo migliore. In molti casi è possibile utilizzare l'evasione come Pietro fa

L'altra opzione è di non usare comprensioni

eggs = (1,3,0,3,2) 

result=[] 
for egg in eggs: 
    try: 
     result.append(egg/0) 
    except ZeroDivisionError: 
     # handle division by zero error 
     # leave empty for now 
     pass 

Fino a voi decidere se questo è più ingombrante o no

Answer 4

mi rendo conto questa domanda è abbastanza vecchio, ma è anche possibile creare una funzione generale di fare questo genere di cose più facili:

def catch(func, handle=lambda e : e, *args, **kwargs): 
    try: 
     return func(*args, **kwargs) 
    except Exception as e: 
     return handle(e) 

Poi, nella vostra comprensione:

eggs = (1,3,0,3,2) 
[catch(lambda : 1/egg) for egg in eggs] 
[1, 0, ('integer division or modulo by zero'), 0, 0] 

Puoi Ovviamente rendi la funzione di default come vuoi (ad esempio, preferisci restituire 'None' per impostazione predefinita).

Spero che questo aiuti voi o eventuali futuri spettatori di questa domanda!

Nota: in python 3, avrei impostato solo la parola chiave argomento "handle" e l'avrei inserita alla fine dell'elenco degli argomenti. Ciò renderebbe molto più naturali le argomentazioni passeggere e simili.

Answer 5

Penso che una funzione di supporto, come suggerito da chi pone la domanda iniziale e Bryan Head, sia buona e non ingombrante. Una singola riga di codice magico che fa tutto il lavoro non è sempre possibile, quindi una funzione di supporto è una soluzione perfetta se si desidera evitare i loop for. Tuttavia vorrei modificarlo per questo:

# A modified version of the helper function by the Question starter 
def spam(egg): 
    try: 
     return 1/egg, None 
    except ZeroDivisionError as err: 
     # handle division by zero error   
     return None, err 

L'output sarà questo [(1/1, None), (1/3, None), (None, ZeroDivisionError), (1/3, None), (1/2, None)]. Con questa risposta hai tutto il controllo per continuare nel modo che preferisci.