Saya ingin mengubah kode berikut

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

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

ke kode ini:

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

Saya mendapatkan kesalahan:

jenis operan yang tidak didukung untuk +: 'generator' dan 'generator'

Bagaimana cara bergabung dengan dua generator di Python?

Saya pikir itertools.chain()harus melakukannya.

Contoh kode:

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

Dengan Python (3,5 atau lebih besar) yang dapat Anda lakukan:

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

Contoh sederhana:

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

Dengan itertools.chain.from_iterable Anda dapat melakukan hal-hal seperti:

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)

Ini dia menggunakan ekspresi generator dengan nested fors:

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]

Satu juga dapat menggunakan operator membongkar *:

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

CATATAN: Bekerja paling efisien untuk iterables 'non-malas'. Dapat juga digunakan dengan berbagai jenis pemahaman. Cara yang disukai untuk concat generator adalah dari jawaban dari @Uduse

Jika Anda ingin membuat generator terpisah tetapi tetap iterate pada mereka pada saat yang sama Anda dapat menggunakan zip ():

CATATAN: Iterasi berhenti di yang lebih pendek dari dua generator

Sebagai contoh:

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

Katakanlah kita harus generator (gen1 dan gen 2) dan kami ingin melakukan beberapa perhitungan tambahan yang membutuhkan hasil dari keduanya. Kita dapat mengembalikan hasil fungsi / perhitungan tersebut melalui metode peta, yang pada gilirannya mengembalikan generator yang bisa kita lewati.

Dalam skenario ini, fungsi / perhitungan perlu diimplementasikan melalui fungsi lambda. Bagian yang sulit adalah apa yang ingin kita lakukan di dalam peta dan fungsi lambda-nya.

Bentuk umum dari solusi yang diusulkan:

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

Semua solusi rumit itu ...

kerjakan saja:

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

Jika Anda benar-benar ingin "bergabung" dengan kedua generator, maka lakukan:

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

Saya akan mengatakan itu, seperti yang disarankan dalam komentar oleh pengguna "wjandrea", solusi terbaik adalah

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

Itu tidak mengubah tipe yang dikembalikan dan benar-benar pythonic.