使用python为网络中的每一对寻找每个可能的短路径

时间:2019-12-12 07:05:54

标签: python

在csv文件中,有一个从父级到子级的有向边列表,如下所示:

    parent  child
0   7       4847454
1   7       14823924
2   7       31208539
3   7       32584634
4   7       54293428

我想找到网络中每对可能的节点之间的最短路径。我怎样才能做到这一点?我不确定哪种包在这种情况下最好(networkX,igraph等)

1 个答案:

答案 0 :(得分:1)

您可以使用networkx(特别是nx.shortest_path)来实现。该算法查找源节点和目标节点之间的最短路径,或者如果未指定任何路径,则生成每个节点对(存在路径的地方)之间所有最短路径的字典。

例如:

>>> df = pd.DataFrame({'parent': {0: 7, 1: 7, 2: 7, 3: 7, 4: 7},
                   'child': {0: 4847454, 1: 14823924, 2: 31208539, 3: 32584634, 4: 54293428}})
>>> G = nx.convert_matrix.from_pandas_edgelist(df, 'parent', 'child', create_using=nx.DiGraph)
>>> nx.shortest_path(G)
{7: {7: [7],
     4847454: [7, 4847454],
     14823924: [7, 14823924],
     31208539: [7, 31208539],
     32584634: [7, 32584634],
     54293428: [7, 54293428]},
 4847454: {4847454: [4847454]},
 14823924: {14823924: [14823924]},
 31208539: {31208539: [31208539]},
 32584634: {32584634: [32584634]},
 54293428: {54293428: [54293428]}}