我想更改下面的代码
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()
我得到了错误:
不支持+:'generator'和'generator'的操作数类型
如何在Python中连接两个生成器?
当前回答
(免责声明:仅限Python 3 !)
与你想要的语法相似的是使用splat操作符展开两个生成器:
for directory, dirs, files in (*os.walk(directory_1), *os.walk(directory_2)):
do_something()
解释:
这有效地执行了将两个生成器的单层扁平化为3元组的n元组(来自os.walk)，如下所示:
((directory1, dirs1, files1), (directory2, dirs2, files2), ...)
然后for循环遍历这个n元组。
当然，通过简单地用括号替换外括号，你可以得到一个3元组的列表，而不是3元组的n元组:
for directory, dirs, files in [*os.walk(directory_1), *os.walk(directory_2)]:
do_something()
结果如下:
[(directory1, dirs1, files1), (directory2, dirs2, files2), ...]
Pro:
这种方法的优点是不需要导入任何东西，也不需要大量代码。
Con:
缺点是将两个生成器转储到一个集合中，然后遍历该集合，有效地进行了两次传递，可能会使用大量内存。
其他回答
代码示例:
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
我想说的是，正如用户“wjandrea”的评论所建议的那样，最好的解决方案是
def concat_generators(*gens):
for gen in gens:
yield from gen
它不会改变返回的类型，并且是真正的python类型。
使用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)
2020年更新:在Python 3和Python 2中都可以工作
import itertools
iterA = range(10,15)
iterB = range(15,20)
iterC = range(20,25)
第一个选项
for i in itertools.chain(iterA, iterB, iterC):
print(i)
# 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
可选选项，在python 2.6中引入
for i in itertools.chain.from_iterable( [iterA, iterB, iterC] ):
print(i)
# 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
Itertools.chain()是基本的。
Itertools.chain.from_iterable()很方便，如果你有一个包含可迭代对象的可迭代对象。例如，每个子目录的文件列表，如[["src/server.py"， "src/readme.txt"]， ["test/test.py"]]]。
