¿Cómo puedo obtener un generador de Python para que devuelva None en lugar de StopIteration?

Question

estoy utilizando generadores para realizar búsquedas en las listas como este sencillo ejemplo:

>>> a = [1,2,3,4] 
>>> (i for i, v in enumerate(a) if v == 4).next() 
3 

(Sólo para enmarcar un poco el ejemplo, estoy usando listas mucho más largos en comparación con la de arriba, y las entradas son una poco más complicado que int. lo hago de esta manera por lo que las listas completas no serán atravesados ​​cada vez que me los busco)

Ahora si me gustaría cambiar eso en lugar de i == 666, devolvería un StopIteration ya que puede' t encuentra cualquier entrada 666 en a.

¿Cómo puedo hacer que vuelva None en su lugar? Por supuesto, podría envolverlo en una cláusula try ... except, pero ¿hay una forma más pitonica de hacerlo?

Answer 1

Si está utilizando Python 2.6+ debe utilizar el next función incorporada, no es el método next (que fue reemplazado por __next__ en 3.x). El next incorporada toma un argumento opcional default volver si se agota el iterador, en lugar de elevar StopIteration:

next((i for i, v in enumerate(a) if i == 666), None) 

Answer 2

se pueden encadenar con el generador (Ninguna,):

from itertools import chain 
a = [1,2,3,4] 
print chain((i for i, v in enumerate(a) if v == 6), (None,)).next() 

pero creo a.index (2) no va a recorrer la lista completa, cuando se encuentra 2, la búsqueda ha terminado. puede probar esto:

>>> timeit.timeit("a.index(0)", "a=range(10)") 
0.19335955439601094 
>>> timeit.timeit("a.index(99)", "a=range(100)") 
2.1938486138533335