ฉันต้องการเปลี่ยนรหัสต่อไปนี้

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

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

รหัสนี้:

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

ฉันได้รับข้อผิดพลาด:

ประเภทตัวถูกดำเนินการที่ไม่รองรับสำหรับ +: 'ตัวกำเนิด' และ 'ตัวสร้าง'

จะเข้าร่วมเครื่องกำเนิดสองเครื่องใน Python ได้อย่างไร

ฉันคิดว่าitertools.chain()ควรทำ

ตัวอย่างของรหัส:

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

ใน Python (3.5 หรือสูงกว่า) คุณสามารถทำได้:

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

ตัวอย่างง่ายๆ:

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

ด้วย itertools.chain.from_iterable คุณสามารถทำสิ่งต่าง ๆ เช่น:

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)

นี่คือการใช้นิพจน์ตัวสร้างกับ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]

ท่านสามารถใช้ตัวดำเนินการ unpack ได้*:

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

หมายเหตุ: ทำงานได้อย่างมีประสิทธิภาพมากที่สุดสำหรับ iterables 'non-lazy' สามารถใช้กับความเข้าใจที่แตกต่างกันได้ วิธีที่ต้องการสำหรับ concat ของเครื่องกำเนิดไฟฟ้าจะมาจากคำตอบจาก@Uduse

หากคุณต้องการให้เครื่องกำเนิดไฟฟ้าแยกจากกัน แต่ยังคงวนซ้ำในเวลาเดียวกันคุณสามารถใช้ zip ():

หมายเหตุ: การวนซ้ำจะหยุดลงเมื่อเครื่องกำเนิดไฟฟ้าทั้งสองสั้นลง

ตัวอย่างเช่น:

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

ให้บอกว่าเราต้องกำเนิด (gen1 และ gen 2) และเราต้องการทำการคำนวณพิเศษบางอย่างที่ต้องการผลลัพธ์ของทั้งคู่ เราสามารถส่งคืนผลลัพธ์ของฟังก์ชัน / การคำนวณดังกล่าวผ่านวิธีการแมปซึ่งจะส่งคืนตัวกำเนิดที่เราสามารถวนซ้ำ

ในสถานการณ์นี้จำเป็นต้องใช้ฟังก์ชัน / การคำนวณผ่านฟังก์ชัน lambda ส่วนที่ยากคือสิ่งที่เราตั้งใจทำในแผนที่และฟังก์ชั่นแลมบ์ดา

รูปแบบทั่วไปของโซลูชันที่เสนอ:

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

โซลูชั่นที่ซับซ้อนทั้งหมดเหล่านั้น ...

แค่ทำ:

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

หากคุณต้องการที่จะ "เข้าร่วม" ทั้งเครื่องปั่นไฟจริงๆให้ทำ:

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

ฉันจะบอกว่าตามที่แนะนำในความคิดเห็นโดยผู้ใช้ "wjandrea" ทางออกที่ดีที่สุดคือ

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

ไม่เปลี่ยนประเภทที่ส่งคืนและเป็น pythonic จริงๆ