Python NetworkX-从嵌套词典中获取DiGraph

时间:2018-08-29 18:38:23

标签: python graph networkx digraphs

我正在尝试从Python的嵌套字典中在python的NetworkX中创建一个简单的DiGraph,但似乎内置的初始化未构建最终的叶子节点。

玩具示例:

class_hierarchy= {-1: ["A", "B"], 
"A":{"A1":[1], "A2":[3,4]}, 
"B": {"B1":[5,6], "B2": [7,8]}}

建筑图:

G = DiGraph(class_hierarchy)

现在让我们看看其中有什么:

G.nodes
Out[86]: NodeView((-1, 'A', 'B', 'A1', 'A2', 'B1', 'B2'))

看起来好像没有添加最终节点

检查:

list(G.successors('A'))
Out[88]: ['A1', 'A2']

看起来很合理

但是:

list(G.successors('A1'))
Out[89]: []

我不确定为什么会这样吗?用于NetworkX的Documentation指定:

incoming_graph_data(输入图(可选,默认值:无))–初始化图的数据。如果为None(默认),则创建一个空图。数据可以是to_networkx_graph()函数支持的任何格式,当前包括边列表, dict ,列表的dict等。

知道我在做什么错吗?

1 个答案:

答案 0 :(得分:1)

您有一个既是dict of lists又是dict of dicts的混合输入。 Networkx会将其解释为dict of lists
请参见following code,其中dataclass_hierarchy

if isinstance(data, dict):
        try:
            #this will raise an exception
            return from_dict_of_dicts(data, create_using=create_using,
                                  multigraph_input=multigraph_input)
        except:
            try:
                # this is what is called in your case
                return from_dict_of_lists(data, create_using=create_using)
            except:
                raise TypeError("Input is not known type.")

在您的情况下,networkx需要一个包含列表邻接表示形式的字典。
例如,预期输入的格式为:key: value-> node u: list of nodes [v1,v2,...,vn] u is connected with(例如{0:[1,2],1:[3,4]}。

您输入的networkx内容如下:

G=nx.DiGraph()

edges_list = [((node, nbr)) for node, nbrlist in d.items() for nbr in nbrlist]
# [(-1, 'A'), (-1, 'B'), ('A', 'A1'), ('A', 'A2'), ('B', 'B1'), ('B', 'B2')]

G.add_edges_from(edges_list)

因此,您必须根据输入的含义更改格式。