python - 惯用的 Python : Propagating yields or flattening sequences?

Question

我正在写一个 breadth 深度优先树遍历函数，我想做的是:

def traverse(node):
    yield node
    for n in node.children:
        yield_all traverse(n) # << if Python had a yield_all statement

这个想法是在树中以一个(平坦的)节点序列结束。

方法 #1:(传播 yield )

def traverse(node):
    yield node
    for n in node.children:
        for m in traverse(n):
            yield m

方法 #2:(展平序列)

def traverse(node):
    return itertools.chain([node],*(traverse(n) for n in node.children))

第一种方法看起来更干净，但我觉得在每个级别显式yield子树中的每个节点很奇怪。

第二种方法简洁且有点脏，但它符合我在 Haskell 中编写的内容:

traverse node = node : concatMap traverse (children node)

所以我的问题是:哪个更好？还是我错过了最佳的第三个选项？

Answer 1

[更新]参见PEP-380 , 此yield all 语法从Python 3.3 开始可用作为 yield from:

def traverse(node):
    yield node
    for n in node.children:
        yield from traverse(n)