single_source_dijkstra#

single_source_dijkstra(G, source, target=None, cutoff=None, weight='weight')[source]#

从源节点找到最短加权路径和长度。

计算加权图中源节点到所有其他可达节点的最短路径长度。

使用Dijkstra算法计算加权图中源节点到所有其他可达节点的最短路径和长度。

Parameters:
GNetworkX图
source节点标签

路径的起始节点

target节点标签, 可选

路径的结束节点

cutoff整数或浮点数, 可选

搜索停止时的长度(边权重的总和)。 如果提供了cutoff,则只返回总权重 <= cutoff的路径。

weight字符串或函数

如果这是一个字符串,则将通过具有此键的边属性访问边权重(即,连接 uv 的边的权重将是 G.edges[u, v][weight] )。如果没有这样的边属性存在,则假设边的权重为1。

如果这是一个函数,则边的权重是该函数返回的值。该函数必须接受三个位置参数:一条边的两个端点和该边的边属性字典。该函数必须返回一个数字或None以表示隐藏的边。

Returns:
distance, path两个字典的元组,或数字和列表。

如果target为None,则计算到所有节点的路径和长度。返回值是两个字典的元组,按目标节点键控。第一个字典存储到每个目标节点的距离。第二个存储到每个目标节点的路径。 如果target不为None,则返回一个元组(distance, path),其中distance是源节点到目标节点的距离,path是一个列表,表示从源节点到目标节点的路径。

Raises:
NodeNotFound

如果 source 不在 G 中。

Notes

边权重属性必须是数值。 距离计算为遍历的加权边之和。

权重函数可以用于通过返回None来隐藏边。因此, weight = lambda u, v, d: 1 if d['color']=="red" else None 将找到最短的红色路径。

基于Python cookbook配方(119466)在 https://code.activestate.com/recipes/119466/

如果边权重为负数或浮点数,此算法不保证能工作(溢出和舍入误差可能导致问题)。

Examples

>>> G = nx.path_graph(5)
>>> length, path = nx.single_source_dijkstra(G, 0)
>>> length[4]
4
>>> for node in [0, 1, 2, 3, 4]:
...     print(f"{node}: {length[node]}")
0: 0
1: 1
2: 2
3: 3
4: 4
>>> path[4]
[0, 1, 2, 3, 4]
>>> length, path = nx.single_source_dijkstra(G, 0, 1)
>>> length
1
>>> path
[0, 1]