Tôi muốn thay đổi mã sau đây

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

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

mã này:

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

Tôi nhận được lỗi:

(các) loại toán hạng không được hỗ trợ cho +: 'trình tạo' và 'trình tạo'

Làm thế nào để tham gia hai trình tạo trong Python?

Tôi nghĩ itertools.chain()nên làm điều đó.

Một ví dụ về mã:

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

Trong Python (3,5 trở lên) bạn có thể làm:

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

Ví dụ đơn giản:

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

Với itertools.chain.from_iterable, bạn có thể làm những việc như:

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)

Ở đây nó đang sử dụng một biểu thức trình tạo với fors lồng nhau :

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]

Người ta cũng có thể sử dụng toán tử giải nén *:

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

LƯU Ý: Hoạt động hiệu quả nhất cho các lần lặp 'không lười biếng'. Cũng có thể được sử dụng với các loại hiểu khác nhau. Cách ưa thích cho concat máy phát điện sẽ là từ câu trả lời từ @Uduse

Nếu bạn muốn giữ các bộ tạo riêng biệt nhưng vẫn lặp lại trên chúng cùng một lúc, bạn có thể sử dụng zip ():

LƯU Ý: Lặp lại dừng ở ngắn hơn của hai máy phát

Ví dụ:

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

Hãy nói rằng chúng ta phải tạo máy phát điện (gen1 và gen 2) và chúng tôi muốn thực hiện một số tính toán bổ sung đòi hỏi kết quả của cả hai. Chúng ta có thể trả về kết quả của hàm / phép tính như vậy thông qua phương thức bản đồ, lần lượt trả về một trình tạo mà chúng ta có thể lặp lại.

Trong kịch bản này, hàm / tính toán cần được thực hiện thông qua hàm lambda. Phần khó khăn là những gì chúng tôi nhắm đến để làm bên trong bản đồ và chức năng lambda của nó.

Hình thức chung của giải pháp đề xuất:

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

Tất cả những giải pháp phức tạp đó ...

cứ làm đi:

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

Nếu bạn thực sự muốn "tham gia" cả hai máy phát điện, thì hãy làm:

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