在Python中使用父/子列表构建树

时间:2013-04-25 00:00:05

标签: python algorithm tree

使用Python,我有一个字典对象列表,其中包含我想要构建到树中的父/子关系。例如:

{'UI': 'T071', 'NAME': 'Entity', 'PARENT': None, 'CHILDREN': 'Conceptual Entity'}
{'UI': 'T077', 'NAME': 'Conceptual Entity', 'PARENT': 'Entitity', 'CHILDREN': 'Organism Attribute, Finding, Idea or Concept'}
{'UI': 'T032', 'NAME': 'Organism Attribute', 'PARENT': 'Conceptual Entity', 'CHILDREN': 'Clinical Attribute'}
etc.

数据集中总共有4个根节点('PARENT'设置为None),它们构成4个独立的树。所以,我打算制作一份树木清单。

数据不一定是任何类型的排序(因此层次结构中较高的节点在列表中不一定更高)。此外,id(UI)没有特定的顺序(树中的T071不一定高于T072)。它们的名称是唯一的,数据集使用它们的名称而不是id(UI)来显示关系。

我有这个简单的课程:

class node():
    def __init__(self, value):
        self.value = value
        self.children = []

    def add_child(self, obj):
        self.children.append(obj)

我对如何解决这个问题感到有点困惑。建议非常感谢。

3 个答案:

答案 0 :(得分:2)

我认为最好是做两次通过。首先,创建一个将名称链接到节点的字典。然后,您可以有效地添加您的项目。

我的代码:

nodes = dict((e["NAME"], node(e)) for e in l)
for e in l:
    if e["PARENT"] is not None:
        nodes[e["PARENT"]].add_children(nodes[e["NAME"])

如果你想要根,你可以使用上面的if,或者你可以过滤节点。

roots = [n for n in nodes.values() if d.value["PARENT"] is None]

答案 1 :(得分:1)

我曾用一个dict代表一个* ix进程树,并为每个父pid代表一个子进程pid列表。所以你得到:

dict_[1] = [2, 3, 4]
dict_[2] = [5, 100]
dict_[3] = [6, 200]
dict_[4] = [7, 300]
dict_[6] = [400]

它看起来效果很好。

可选择是否希望叶节点与空列表一起存在,或者只是不在树中显示。我已经在上面显示它们没有出现在dict级别的树中。

我认为这只适用于pid(节点)只能出现在树中的一个位置。 EG,100不能是2 4的孩子。

答案 2 :(得分:-2)

代码深入字典树多达 5 层。当到达某个对象时,它会检查它是否是 numpy。如果是,则打印其形状。如果不是,则写入对象的类型。通过选择“级别”,它可以确定树应该走多深。当前的实现最多可达 5 个级别。

def tree_view_dict(d, level=999):
    for key in d.keys():
        print('|')
        print(key, end = '')
        if isinstance(d[key], dict) and level > 1:
            print()
            for key2 in d[key].keys():
                print('|__', key2, end = '')
                if isinstance(d[key][key2], dict) and level > 2:
                    print()
                    for key3 in d[key][key2]:
                        print('   |__', key3, end = '')
                        if isinstance(d[key][key2][key3], dict) and level > 3:
                            print()
                            for key4 in d[key][key2][key3]:
                                print('     |__', key4, end = '')
                                if isinstance(d[key][key2][key3][key4], dict) and level > 4:
                                    print()
                                    for key5 in d[key][key2][key3][key4]:
                                        print('       |__', key5, end = '')
                                else:
                                    print(': ', d[key][key2][key3][key4].shape if type(d[key][key2][key3][key4]).__module__ == 'numpy' else type(d[key][key2][key3][key4]))
                        else:
                            print(': ', d[key][key2][key3].shape if type(d[key][key2][key3]).__module__ == 'numpy' else type(d[key][key2][key3]))
                else:
                    print(': ', d[key][key2].shape if type(d[key][key2]).__module__ == 'numpy' else type(d[key][key2]))
        else:
            print(': ', d[key].shape if type(d[key]).__module__ == 'numpy' else type(d[key]))
    pass

d = {'UI': 'T032', 'NAME': 'Organism Attribute', 'PARENT': 'Conceptual Entity', 'CHILDREN': {'CHILDREN_array': np.array([[1,2], [3,4]]),'CHILDREN_name': 'name'}}
tree_view_dict(d, level=5)

结果为,


|
UI:  <class 'str'>
|
NAME:  <class 'str'>
|
PARENT:  <class 'str'>
|
CHILDREN
|__ CHILDREN_array:  (2, 2)
|__ CHILDREN_name:  <class 'str'>```