Come unire due generatori in Python?

Question

Voglio cambiare il seguente codice

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

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

a questo codice:

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

ottengo l'errore:

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

Come unire due generatori in Python?

Answer 1

Penso che itertools.chain() dovrebbe farlo.

Answer 2

Un esempio di codice:

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

Con itertools.chain.from_iterable si possono fare cose come:

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

semplice esempio:

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

Se si desidera mantenere i generatori si separano ma continuano a scorrere su di essi allo stesso tempo puoi usare zip():

NOTA: iterazione si ferma al più corto dei due generatori

Ad esempio:

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

Diciamo che dobbiamo generatori (Gen1 e Gen 2) e vogliamo eseguire alcuni calcoli in più che richiede il risultato di entrambi. Possiamo restituire il risultato di tale funzione/calcolo attraverso il metodo map, che a sua volta restituisce un generatore su cui possiamo eseguire il looping.

In questo scenario, la funzione/calcolo deve essere implementata tramite la funzione lambda. La parte difficile è ciò che miriamo a fare all'interno della mappa e la sua funzione lambda.

forma generale della soluzione proposta:

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

Answer 7

In python3 + si può fare:

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