Python 生成器：递归树的正确代码

Question

class Node(object):
    def __init__(self, lst):
        if type(lst) == list:
            self.value = lst[0]
            self.children = lst[1:]
        else:
            self.value = lst
            self.children = []
    @property
    def ChildElements(self):
        return [Node(a) for a in self.children]

    @property
    def GetValue(self):
        return self.value

def node_recurse_generator(node):
    yield node.value
    for n in node.ChildElements:
        node_recurse_generator(n)

节点是一种简单的树状数据结构。列表的第一个元素始终是节点的值，其他元素将是子元素。如果 Node 是用列表以外的东西启动的，则该值将是该值，并且子节点将是 []，一个空列表。

a = Node([1,[10,20,30],[100,200,300]])
>>> list(node_recurse_generator(a))
[1]

现在，如果对所有元素都有一个递归迭代器就太好了，但我的生成器只生成第一个元素。为什么会发生这种情况？

Answer 1

简单地递归调用 node_recurse_generator 是不够的 - 您必须yield其结果：

def node_recurse_generator(node):
    yield node.value
    for n in node.ChildElements:
        for rn in node_recurse_generator(n):
            yield rn

Answer 2

正如顶部答案的评论中提到的 yak ，您还可以在 Python 之后使用 yield from 3.3.

def node_recurse_generator(node):
    yield node.value
    for n in node.ChildElements:
        yield from node_recurse_generator(n)