Besoin d'ajouter un élément au début d'un itérateur en python

Question

J'ai un programme comme suit:

a=reader.next() 
if *some condition holds*: 
    #Do some processing and continue the iteration 
else: 
    #Append the variable a back to the iterator 
    #That is nullify the operation *a=reader.next()* 

Comment puis-je ajouter un élément au début de la iterator? (Ou existe-t-il un moyen plus facile de faire cela?)

EDIT: OK laissez-moi le mettre ainsi. J'ai besoin de l'élément suivant dans un itérateur sans le supprimer. Comment procéder?>?

Answer 1

Les itérateurs Python, en tant que tels, ont des fonctionnalités très limitées - pas d'ajout ou quelque chose comme ça. Vous devrez envelopper l'itérateur générique dans un wrapper en ajoutant cette fonctionnalité. .: par exemple

class Wrapper(object): 
    def __init__(self, it): 
    self.it = it 
    self.pushedback = [] 
    def __iter__(self): 
    return self 
    def next(self): 
    if self.pushedback: 
     return self.pushedback.pop() 
    else: 
     return self.it.next() 
    def pushback(self, val): 
    self.pushedback.append(val) 

Ceci est Python 2.5 (devrait fonctionner en 2.6 aussi) - de légères variantes conseillées pour 2,6 et obligatoire pour 3.Toute (utiliser next(self.it) au lieu de self.it.next() et de définir __next__ au lieu de next).

Édition: l'OP dit maintenant ce dont il a besoin: "Regarder en avant sans consommer". Emballage est toujours la meilleure option, mais une alternative est:

import itertools 
    ... 
o, peek = itertools.tee(o) 
if isneat(peek.next()): ... 

ce n'avance o (rappelez-vous avancer si et quand vous décidez de ne voulez ;-).

Answer 2

De par leur conception (dans les concepts généraux de développement), les itérateurs sont destinés à être en lecture seule, et toute tentative de les changer se casserait. Alternativement, vous pouvez lire l'itérateur en arrière, et l'ajouter à la fin de l'élément hte (qui est en fait le début :) :)?

Answer 3

Ce n'est pas trop proche de ce que vous avez demandé, mais si vous avez le contrôle sur le générateur et que vous n'avez pas besoin de jeter un coup d'œil avant que la valeur ne soit générée, vous pouvez utiliser la méthode generator.send pour dire le générateur de répéter la dernière valeur qu'il a produit:

>>> def a(): 
...  for x in (1,2,3): 
...    rcvd = yield x 
...    if rcvd is not None: 
...      yield x 
... 
>>> gen = a() 
>>> gen.send("just checking") 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
TypeError: can't send non-None value to a just-started generator 
>>> gen.next() 
1 
>>> gen.send("just checking") 
1 
>>> gen.next() 
2 
>>> gen.next() 
3 
>>> gen.next() 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
StopIteration