Question

它不是一个严格的嵌套列表，它是一个树形结构，如下所示：

A = [a, [b, c,[d,e]]]

和相应的树是：

每当在一个元素之后存在子列表时，子列表对应于该元素的子节点。否则，元素位于同一层。我想生成一个字典，每个节点分别作为一个键，如：

child[a] = [b,c,d,e]
child[c] = [d,e]

我怎么能在python中做到这一点？或者树结构转换还有其他更好的建议吗？

Answer 1

如果您要进行大量的图形操作，我会考虑导入networkx，因为它会使事情变得更容易。要将嵌套列表解析为networkx树：

import networkx as nx

def parse_tree(node_list):
    """Parses a nested list into a networkx tree."""
    tree = nx.DiGraph()
    root = node_list[0]
    tree.add_node(root)

    queue = [(root, node_list[1])]

    while queue:
        parent, nodes = queue.pop(0)
        prev = None
        for node in nodes:
            if isinstance(node, list):
                queue.append((prev, node))
            else:
                tree.add_node(node)
                tree.add_edge(parent, node)

            prev = node

    return tree

使用此函数，可以轻松获取每个节点后代的字典：

>>> l = ["a", ["b", "c",["d","e"]]]
>>> tree = parse_tree(l)
>>> {node: nx.descendants(tree, node) for node in tree}
{'a': {'b', 'c', 'd', 'e'},
 'b': set(),
 'c': {'d', 'e'},
 'd': set(),
 'e': set()}

Answer 2

我仍然认为你应该使用/获得现有实现的启发，但如果你需要这个工作，这可能是你正在寻找的：

#!/usr/bin/env python

# added a test case
B = ['a', ['b', 'c',['d','e']], 'f', ['g', 'h']]
A = ['a', ['b', 'c',['d','e']]]

# found on stack overflow - flatten list of kids for parent
def flatten(iterable):
    """Recursively iterate lists and tuples.
    """
    for elm in iterable:
        if isinstance(elm, (list, tuple)):
            for relm in flatten(elm):
                yield relm
        else:
            yield elm

# add data to an existing tree (recursive)
def treeify(tree, l):
    if isinstance(l, list):
        # avoid looking back
        l.reverse()
    for index in range(len(l)):
        if isinstance(l[index], list):
            parent_name = l[index+1]
            # flatten kids to a list
            tree[parent_name] = list(flatten(l[index]))
            # continue in deeper lists
            treeify(tree, l[index])

tree = {}
treeify(tree, A)
print tree
tree = {}
treeify(tree, B)
print tree

这会使list反转，以避免在遍历它时回头看。如果当前的名称是list，Tt将名称设置为下一个成员，并立即（递归地）遍历子元素。

将嵌套列表转换为python中的字典

2 个答案: