Wie verbinden Sie zwei Generatoren in Python?

Question

Ich möchte

for directory, dirs, files in os.walk(directory_1): 
    do_something() 

for directory, dirs, files in os.walk(directory_2): 
    do_something() 

zu diesem Code den folgenden Code ändern:

for directory, dirs, files in os.walk(directory_1) + os.walk(directory_2): 
    do_something() 

ich den Fehler:

unsupported operand type(s) for +: 'generator' and 'generator'

Wie zwei Generatoren in Python verbinden?

Answer 1

Ich denke, itertools.chain() sollte es tun.

Answer 2

Ein Beispiel für Code:

from itertools import chain 

def generator1(): 
    for item in 'abcdef': 
     yield item 

def generator2(): 
    for item in '123456': 
     yield item 

generator3 = chain(generator1(), generator2()) 
for item in generator3: 
    print item 

Answer 3

Mit itertools.chain.from_iterable Sie können wie die Dinge tun:

def genny(start): 
    for x in range(start, start+3): 
    yield x 

y = [1, 2] 
ab = [o for o in itertools.chain.from_iterable(genny(x) for x in y)] 
print(ab) 

Answer 4

Einfaches Beispiel:

from itertools import chain 
x = iter([1,2,3])  #Create Generator Object (listiterator) 
y = iter([3,4,5])  #another one 
result = chain(x, y) #Chained x and y 

Answer 5

Wenn Sie behalten möchten Die Generatoren trennen sich aber iterieren immer noch über sie zur gleichen Zeit können Sie zip() verwenden:

HINWEIS: Iteration stoppt an der kürzeren der beiden Generatoren

Zum Beispiel:

for (root1, dir1, files1), (root2, dir2, files2) in zip(os.walk(path1), os.walk(path2)): 

    for file in files1: 
     #do something with first list of files 

    for file in files2: 
     #do something with second list of files 

Answer 6

Lassen Sie uns sagen, dass wir zu Generatoren (Gen1 und Gen2) haben, und wir wollen einige zusätzliche Berechnung ausführen zu können, erfordert das Ergebnis von beiden. Wir können das Ergebnis einer solchen Funktion/Berechnung durch die map-Methode zurückgeben, die wiederum einen Generator zurückgibt, den wir durchlaufen können.

In diesem Szenario muss die Funktion/Berechnung über die Lambda-Funktion implementiert werden. Der schwierige Teil ist, was wir innerhalb der Karte und seiner Lambda-Funktion tun wollen.

Allgemeine Form der vorgeschlagenen Lösung:

def function(gen1,gen2): 
     for item in map(lambda x, y: do_somethin(x,y), gen1, gen2): 
      yield item 

Answer 7

In Python3 + können Sie tun:

def concat(a, b): 
    yield from a 
    yield from b