是否可以将元素附加到python生成器?
我目前正在尝试从一组无组织的文件夹中获取所有图像,并将它们写入新目录。为了获取文件,我使用的是os.walk(),它在单个目录中返回一个图像文件列表。虽然我可以从这个单独的列表中创建一个生成器,但我不知道如何将所有这些列表组合成一个单独的生成器。任何帮助将不胜感激。
相关:
答案 0 :(得分:18)
您正在寻找itertools.chain
。它将多个迭代组合成一个,如下所示:
>>> for i in itertools.chain([1,2,3], [4,5,6]):
... print i
...
1
2
3
4
5
6
答案 1 :(得分:14)
这应该这样做,其中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
答案 2 :(得分:4)
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
答案 3 :(得分:-1)
喜欢这个。
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
运作良好。