Wie umleiten Daten in einem Iterator in zwei andere?

Question

Ich weiß, dass ich einen Iterator kopieren

x1, x2 = itertools.tee(x)

Dann wird, um zwei Generatoren zu bekommen verwenden, könnte ich filtern:

filter(..., x1); filter(..., x2)

aber dann würde ich das gleiche läuft Berechnung zweimal, dh gehe durch x in x1 und x2.

So würde ich etwas effizienter wie das tun:

x1, x2 = divert(into x1 if ... else x2, x)

mag dies alles in Python existieren 3?

Answer 1

Es gibt kein integriertes Werkzeug in Python geschrieben, die ich kenne. Es ist ein kleiner Trick, um zu funktionieren, da es keine Garantie für die Aufrufreihenfolge jedes Iterators gibt, den Sie erzeugen können.

Zum Beispiel x könnte einen x1 Wert von einem x2 Wert gefolgt produzieren, aber Ihr Code könnte über x1 laufen, bis er einen Signalwert erzeugt, dann auf x2 laufen, bis er einen Signalwert erzeugt ... Also im Grunde würde der Code müssen alle x2 Werte halten, bis ein x1 Wert generiert wird, der beliebig spät sein kann.

Wenn das wirklich das ist, was Sie tun möchten, hier ist eine kurze Idee, wie man diesen Puffer macht. Warnung, es ist überhaupt nicht getestet und angenommen, x ist ein endloser Generator. Außerdem müssen Sie zwei tatsächliche Iterator-Klassen codieren, die __next__ implementieren, das sich auf diesen allgemeinen Iterator bezieht, einen mit category==True und den anderen mit category==False.

class SeparatedIterator: 
    def __init__(self, iterator, filter): 
     self.it = iterator 
     self.f = filter 
     #The buffer contains pairs of (value,filterIsTrue) 
     self.valueBuffer = [] 

    def generate(): 
     value = next(self.it) 
     filtered = self.f(value) 
     self.valueBuffer.append((value, filtered)) 

    def nextValue(category): 
     #search in stored values 
     for i in range(len(self.valueBuffer)): 
      value, filtered = self.valueBuffer[i] 
      if filtered == category: 
       del self.valueBuffer[i] 
       return value 

     #else, if none of the category found, 
     #generate until one of the category is made 
     self.generate() 
     while self.valueBuffer[-1][1] != category: 
      self.generate() 

     #pop the value and return it 
     value, _ = self.valueBuffer.pop() 
     return value 

Else wenn Sie mehr Kontrolle über den Iterator Anruf Auftrag haben, müssen Sie dieses Wissen nutzen, um eine maßgeschneiderte und optimierte Art und Weise zu implementieren zwischen Iteratoren Werten zu wechseln.