Добавление элементов в генераторы питона

Question

Можно ли добавлять элементы в генератор питона?

В настоящее время я пытаюсь получить все изображения из набора неорганизованных папок и записать их в новый каталог. Чтобы получить файлы, я использую os.walk(), который возвращает список файлов изображений в одном каталоге. Хотя я могу сделать генератор из этого единственного списка, я не знаю, как объединить все эти списки в один генератор. Любая помощь приветствуется.

Похожие:

• Flattening a shallow list in 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

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 3

Как это.

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 

Хорошо работает.

Answer 4

Вы ищете itertools.chain. Он сочетает в себе несколько итерируемых в один, как это:

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