Dodawanie elementów do generatorów pytonów

Question

Czy można dołączyć elementy do generatora pythonów?

Obecnie próbuję uzyskać wszystkie obrazy z zestawu zdezorganizowanych folderów i zapisać je w nowym katalogu. Aby uzyskać pliki, używam os.walk(), który zwraca listę plików obrazów w jednym katalogu. Chociaż mogę utworzyć generator z tej pojedynczej listy, nie wiem jak połączyć wszystkie te listy w jeden generator. Każda pomoc byłaby bardzo cenna.

pokrewne:

• Flattening a shallow list in python

Answer 1

To powinno wystarczyć, gdzie directories jest lista katalogów:

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

jak ten.

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 

Działa dobrze.

Answer 4

Szukasz itertools.chain. Będzie łączyć wiele iterables w jeden, tak:

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