我想更改以下代码

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()

我得到错误:

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

如何在 Python 中加入两个生成器？

最佳答案

itertools.chain()应该这样做。它需要多个迭代器，并从一个一个地产生，大致相当于:

def chain(*iterables):
    for it in iterables:
        for element in it:
            yield element

使用示例:

from itertools import chain

g = (c for c in 'ABC')  # Dummy generator, just for example
c = chain(g, 'DEF')  # Chain the generator and a string
for item in c:
    print(item)

输出:

A
B
C
D
E
F

https://stackoverflow.com/questions/3211041/