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:

tipi di operando non supportati per +: "generatore" e "generatore"

Come unire due generatori in Python?

Penso che itertools.chain()dovrebbe farlo.

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

In Python (3.5 o versioni successive) puoi fare:

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

Esempio semplice:

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

Con itertools.chain.from_iterable puoi 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)

Qui sta usando un'espressione di generatore con fors nidificati :

a = range(3)
b = range(5)
ab = (i for it in (a, b) for i in it)
assert list(ab) == [0, 1, 2, 0, 1, 2, 3, 4]

Si può anche usare l'operatore di decompressione *:

concat = (*gen1(), *gen2())

NOTA: funziona in modo più efficiente per iterabili "non pigri". Può anche essere usato con diversi tipi di comprensione. Il modo preferito per il concat generatore sarebbe dalla risposta di @Uduse

Se vuoi mantenere separati i generatori, ma continua a iterarli contemporaneamente puoi usare zip ():

NOTA: L'iterazione si arresta sul più corto dei due generatori

Per 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

Diciamo che dobbiamo generatori (gen1 e gen 2) e vogliamo eseguire alcuni calcoli extra che richiedono il risultato di entrambi. Possiamo restituire il risultato di tale funzione / calcolo attraverso il metodo della mappa, che a sua volta restituisce un generatore su cui possiamo eseguire il loop.

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

Tutte quelle soluzioni complicate ...

basta fare:

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

Se vuoi davvero "unire" entrambi i generatori, allora fai:

for directory, dirs, files in 
        [x for osw in [os.walk(director_1), os.walk(director_2)] 
               for x in osw]:
    do_something()

Direi che, come suggerito nei commenti dell'utente "wjandrea", la soluzione migliore è

def concat_generators(*args):
    for gen in args:
        yield from gen

Non cambia il tipo restituito ed è veramente pythonic.