Adding elements to python generators

前端 未结 4 1625
醉酒成梦
醉酒成梦 2020-12-13 19:56

Is it possible to append elements to a python generator?

I\'m currently trying to get all images from a set of disorganized folders and write them to a new directory

4条回答
  •  孤街浪徒
    2020-12-13 20:49

    This should do it, where directories is your list of 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
    

提交回复
热议问题