Question

我在Python中有一个带有加权边的networkx图。我想获得两个节点之间最小路径的权重。

当前，我从nx.shortest_path实现中获得最短路径中的节点，然后遍历每对节点并求和每对节点之间的权重。

shortest_path = nx.shortest_path(G, source, destination, 'distance')

#function to iterate over each pair

import itertools
def pairwise(iterable):
    a, b = itertools.tee(iterable)
    next(b, None)
    return zip(a, b)

weightSum = 0
for adjPair in pairwise(shortest_path):
    weightSum = weightSum + G[adjPair[0]][adjPair[1]]['distance']

是否有更好的（内置）替代方法？

Answer 1

您寻找single_source_dijkstra：

from networkx.algorithms.shortest_paths.weighted import single_source_dijkstra

single_source_dijkstra(G,s,t)

示例

import networkx as nx
from networkx.algorithms.shortest_paths.weighted import single_source_dijkstra

G = nx.Graph()

G.add_edge('a', 'b', weight=0.6)
G.add_edge('a', 'c', weight=6)
G.add_edge('c', 'd', weight=0.1)
G.add_edge('c', 'e', weight=0.7)
G.add_edge('c', 'f', weight=0.9)
G.add_edge('a', 'd', weight=0.3)

single_source_dijkstra(G,'b','f')

输出

(1.9, ['b', 'a', 'd', 'c', 'f'])

Answer 2

networkx文档的页面为：shortest paths。

有几种选择，但看起来shortest_path_length()就是您想要的。

为清楚起见：

shortest_path = nx.shortest_path_length(G, source, destination, 'distance')

如何获得两个节点之间最小路径的权重？

2 个答案: