trova la prima sequenza che corrisponde a un criterio

Question

Quale sarebbe il modo più elegante ed efficiente per trovare/restituire la prima voce dell'elenco che corrisponde a un determinato criterio?

Ad esempio, se ho un elenco di oggetti e vorrei ottenere il primo oggetto di quelli con attributo obj.val==5. Naturalmente potrei usare la comprensione delle liste, ma ciò dovrebbe comportare O (n) e se n è grande, è uno spreco. Potrei anche usare un ciclo con break una volta che il criterio è stato rispettato, ma ho pensato che ci potrebbe essere una soluzione più pitone/elegante.

Answer 1

Se non si dispone di altri indici o le informazioni ordinate per gli oggetti, quindi si dovrà scorrere fino a quando viene trovato un tale oggetto:

next(obj for obj in objs if obj.val==5) 

Questo è comunque più veloce di una lista completa comprensione. Confronta questi due:

[i for i in xrange(100000) if i == 1000][0] 

next(i for i in xrange(100000) if i == 1000) 

La prima si ha la necessità 5.75ms, la seconda 58.3μs (100 volte più veloce perché il ciclo 100 volte più breve).

Answer 2

a=[100,200,300,400,500] 
def search(b): 
try: 
    k=a.index(b) 
    return a[k] 
except ValueError: 
    return 'not found' 
print(search(500)) 

che restituisca l'oggetto se trovato altrimenti tornerà "non trovato"