我想更改下面的代码
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中连接两个生成器?
当前回答
你也可以使用解包操作符*:
concat = (*gen1(), *gen2())
注意:对于“非惰性”迭代对象最有效。也可以用于不同类型的推导式。生成器concat的首选方式将来自@Uduse的答案
其他回答
代码示例:
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
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
假设我们有两个生成器(第1代和第2代)，我们想要执行一些额外的计算，需要这两个生成器的结果。 我们可以通过map方法返回这种函数/计算的结果，该方法反过来返回一个生成器，我们可以循环使用它。
在这种情况下，函数/计算需要通过lambda函数实现。 棘手的部分是我们打算在映射及其lambda函数中做什么。
建议解决方案的一般形式:
def function(gen1,gen2):
for item in map(lambda x, y: do_somethin(x,y), gen1, gen2):
yield item
如果你想保持生成器的分离，但仍然在同一时间遍历它们，你可以使用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
(免责声明:仅限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:
缺点是将两个生成器转储到一个集合中，然后遍历该集合，有效地进行了两次传递，可能会使用大量内存。
