Récupère le nième élément d'un générateur en Python

Question

Existe-t-il une manière plus concise et plus synthétique d'écrire ce qui suit?

gen = (i for i in xrange(10)) 
index = 5 
for i, v in enumerate(gen): 
    if i is index: 
     return v 

Il semble presque naturel qu'un générateur doit avoir une expression gen[index], qui agit comme une liste, mais est fonctionnellement identique au code ci-dessus.

Answer 1

une méthode serait d'utiliser itertools.islice

>>> next(itertools.islice(xrange(10), 5, 5 + 1)) 
5 

Answer 2

Peut-être devriez-vous élaborer davantage sur un cas d'utilisation réel.

>>> gen = xrange(10) 
>>> ind=5 
>>> gen[ind] 
5 

Answer 3

Vous pouvez le faire, en utilisant count comme générateur d'exemple:

from itertools import islice, count 
next(islice(count(), n, n+1)) 

Answer 4

vous pouvez simplement convertir le générateur dans une liste et utiliser l'index comme d'habitude:

>>> [i for i in range(10)][index] 
5 

Answer 5

Je discuterais contre la tentation de traiter les générateurs comme des listes. L'approche simple mais naïve est le simple-liner:

gen = (i for i in range(10)) 
list(gen)[3] 

Mais rappelez-vous, les générateurs ne sont pas comme les listes. Ils ne stockent leurs résultats intermédiaires nulle part, donc vous ne pouvez pas revenir en arrière. Je démontrerai le problème avec un exemple simple dans le rempl python:

>>> gen = (i for i in range(10)) 
>>> list(gen)[3] 
3 
>>> list(gen)[3] 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
IndexError: list index out of range 

Une fois que vous commencez à passer par un générateur pour obtenir la valeur n-ième dans la séquence, le générateur est maintenant dans un état différent, et en essayant de obtenir la nième valeur à nouveau vous retournera un résultat différent, ce qui entraînera probablement un bug dans votre code.

Jetons un coup d'oeil à un autre exemple, basé sur le code de la question.

On s'attendait initialement à ce que les éléments suivants impriment deux fois 4.

gen = (i for i in range(10)) 
index = 4 
for i, v in enumerate(gen): 
    if i == index: 
     answer = v 
     break 
print(answer) 
for i, v in enumerate(gen): 
    if i == index: 
     answer = v 
     break 
print(answer) 

mais tapez ceci dans le rempl et vous obtenez:

>>> gen = (i for i in range(10)) 
>>> index = 4 
>>> for i, v in enumerate(gen): 
...  if i == index: 
...    answer = v 
...    break 
... 
>>> print(answer) 
4 
>>> for i, v in enumerate(gen): 
...  if i == index: 
...    answer = v 
...    break 
... 
>>> print(answer) 
9 

Bonne chance que le traçage bug vers le bas.

Answer 6

La première chose qui vient à mon esprit était:

gen = (i for i in xrange(10)) 
index = 5 

for i, v in zip(range(index), gen): pass 

return v