在Python中生成字典树中的所有叶到根路径

Question

我有一个"非标准"形式的字典树,如下所示:

tree = {'0': {'A': {'B': {'C': {}}}},
             {'D': {'E': {}},
                   {'F': {}}}}

叶节点被定义为字典键值对,其中值是空字典.我想提取所有叶到根路径作为列表列表,如下所示:

paths_ = [['C', 'B', 'A', '0'],
          ['E', 'D', '0'],
          ['F', 'D', '0']]

如果有用,路径也可以反转.

paths_ = [['0', 'A', 'B', 'C'],
          ['0', 'D', 'E'],
          ['0', 'D', 'F']]

我知道我必须递归地做,我需要每个路径的累加器列表.如果函数产生路径列表也会很好.到目前为止我所拥有的是:

def paths(node, subtree, acc=[]):
    if not subtree:
        yield [node]+acc
    for n, s in subtree.items():
        yield paths(n, s, acc)

它并没有真正做我想要的:

paths_ = list(paths('0', tree['0']))

理想情况下,这应该返回列表列表.任何帮助都感激不尽.

Answer 1

假设您实际上打算使用以下结构tree:

tree = {'0': {'A': {'B': {'C': {}}},
              'D': {'E': {},
                    'F': {}}}}

这是一个类似的paths()功能,应该做你想要的:

def paths(tree, cur=()):
    if not tree:
        yield cur
    else:
        for n, s in tree.items():
            for path in paths(s, cur+(n,)):
                yield path

结果:

>>> list(paths(tree))
[('0', 'A', 'B', 'C'), ('0', 'D', 'E'), ('0', 'D', 'F')]

请注意,我使用元组作为默认参数而不是列表,这是因为可变的默认参数可能会让您遇到麻烦.