向 python 生成器添加元素

Question

是否可以将元素附加到 python 生成器？

我目前正在尝试从一组杂乱的文件夹中获取所有图像并将它们写入一个新目录。为了获取文件，我使用 os.walk() 它返回单个目录中的图像文件列表。虽然我可以用这个列表创建一个生成器，但我不知道如何将所有这些列表组合成一个生成器。任何帮助将非常感激。

有关的：

• 在 python 中展平浅列表

Answer 1

这应该可以做到，在哪里 directories 是您的目录列表：

import os
import itertools

generators = [os.walk(d) for d in directories]
for root, dirs, files in itertools.chain(*generators):
    print root, dirs, files

Answer 2

您正在寻找 itertools.chain. 。它将多个迭代合并为一个迭代，如下所示：

>>> for i in itertools.chain([1,2,3], [4,5,6]):
...  print i
... 
1
2
3
4
5
6

Answer 3

def files_gen(topdir='.'):
    for root, dirs, files in os.walk(topdir):
        # ... do some stuff with files
        for f in files:
            yield os.path.join(root, f)
        # ... do other stuff

for f in files_gen():
    print f

Answer 4

像这样。

def threeGens( i, j, k ):
    for x in range(i):
       yield x
    for x in range(j):
       yield x
    for x in range(k):
       yield x

效果很好。