Python：懒惰迭代通过类似目录的结构

Question

我想通过类似目录的结构来应用函数。由于Python 3支持懒惰的生成器，我想过用这些来懒惰地评估结果列表。这是我的做法：

iterate = lambda func, d: (func(f) for f in listDir(d) if not f.isDir) + sum((iterate(func, d + f.name) for f in listDir(d) if f.isDir.), []) 

这导致unsupported operand type(s) for +: 'generator' and 'list'。有懒惰的发电机来表达这种习惯用法吗？

listDir接受一个路径并返回一个文件信息列表。

Answer 1

假设的顺序并不重要，我认为这将通常这样写：

def iterate(func, d): 
    for f in listDir(d): 
     if f.isDir: 
      yield from iterate(func, d + f.name) 
     else: 
      yield func(f) 

yield from只有在Python 3.3存在，在此之前，你可以这样做：

for x in iterate(func, d + f.name): 
    yield x 

如果您需要将发生器串联在一起，然后itertools.chain和itertools.chain.from_iterator是作业的标准工具：

def iterate(func, d): 
    return itertools.chain(
     (func(f) for f in listDir(d) if not f.isDir), 
     itertools.chain.from_iterable(
      iterate(func, d + f.name) for f in listDir(d) if f.isDir) 
     ) 
    )