如何在 Python 中动态构建树

Question

一个初学者 Python/编程问题...我想在 Python 中构建一个树形结构，最好基于字典。我发现代码可以巧妙地做到这一点：

Tree = lambda: collections.defaultdict(Tree)
root = Tree()

这可以很容易地填充为：

 root['toplevel']['secondlevel']['thirdlevel'] = 1
 root['toplevel']['anotherLevel'] = 2
 ...etc.

我想动态填充关卡/叶子，以便我可以根据需要添加任意多的关卡，并且叶子可以在任何关卡的位置。我该怎么做？

非常感谢任何帮助。

Answer 1

这个

root['toplevel']['secondlevel']['thirdlevel'] = 1

也可以这样：

node = root
for key in ('toplevel', 'secondlevel'):
    node = node[key]
node['thirdlevel'] = 1

我希望这能给你一个想法。

Answer 2

啊！当我开始编码时，这对我来说也是一个问题，但我们中最优秀的人很早就遇到了。

注意；这适用于当你的树深入 N 层时。其中 N 介于 0 和无穷大之间，即；你不知道它能走多深；它可能只有第一级，也可能达到第 20 级

您的问题是一般编程问题；读一棵树，可以是任意数量的深度，解决方案是；递归。

无论何时在树结构中阅读，你都必须；

1 - 建立一个对象 2 - 检查对象是否有孩子 2a - 如果对象有子对象，则为每个子对象执行步骤 1 和 2。

这里有一个用于执行此操作的 python 代码模板；

def buildTree(treeObject):
   currObject = Hierarchy()
   currObject.name = treeObject.getName()
   currObject.age = treeObject.getAge()
   #as well as any other calculations and values you have to set for that object

   for child in treeObject.children:
      currChild = buildTree(child)
      currObject.addChild(currChild)
   #end loop

   return currObject

Answer 3

你可以简单地用一个实用函数来做，像这样

def add_element(root, path, data):
    reduce(lambda x, y: x[y], path[:-1], root)[path[-1]] = data

你可以像这样使用它

import collections
tree = lambda: collections.defaultdict(tree)
root = tree()
add_element(root, ['toplevel', 'secondlevel', 'thirdlevel'], 1)
add_element(root, ['toplevel', 'anotherlevel'], 2)
print root

输出

defaultdict(<function <lambda> at 0x7f1145eac7d0>,
    {'toplevel': defaultdict(<function <lambda> at 0x7f1145eac7d0>,
       {'secondlevel': defaultdict(<function <lambda> at 0x7f1145eac7d0>,
            {'thirdlevel': 1}),
        'anotherlevel': 2
       })
    })

如果你想以递归的方式实现，你可以从当前root获取第一个元素并获取子对象，然后从path中剥离第一个元素，用于下一次迭代。

def add_element(root, path, data):
    if len(path) == 1:
        root[path[0]] = data
    else:
        add_element(root[path[0]], path[1:], data)