當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python NetworkX single_source_dijkstra用法及代碼示例


本文簡要介紹 networkx.algorithms.shortest_paths.weighted.single_source_dijkstra 的用法。

用法:

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

從源節點查找最短的加權路徑和長度。

計算加權圖的源和所有其他可達節點之間的最短路徑長度。

使用 Dijkstra 算法計算源與加權圖中所有其他可到達節點之間的最短路徑和長度。

參數

GNetworkX 圖
source節點標簽

路徑的起始節點

target節點標簽,可選

路徑的結束節點

cutoff整數或浮點數,可選

停止搜索的長度(邊權重的總和)。如果提供了截止,則僅返回總權重 <= 截止的路徑。

weight字符串或函數

如果這是一個字符串,則將通過帶有此鍵的邊屬性訪問邊權重(即,連接 uv 的邊的權重將為 G.edges[u, v][weight] )。如果不存在這樣的邊屬性,則假設邊的權重為 1。

如果這是一個函數,則邊的權重是函數返回的值。該函數必須準確地接受三個位置參數:一條邊的兩個端點和該邊的邊屬性字典。該函數必須返回一個數字。

返回

distance, path一對字典,或數字和列表。

如果目標為無,則計算所有節點的路徑和長度。返回值是由目標節點鍵入的兩個字典的元組。第一個字典存儲到每個目標節點的距離。第二個存儲到每個目標節點的路徑。如果目標不是無,則返回一個元組(距離,路徑),其中距離是源到目標的距離,路徑是表示從源到目標的路徑的列表。

拋出

NodeNotFound

如果 source 不在 G 中。

注意

邊權重屬性必須是數字。距離計算為遍曆的加權邊的總和。

權重函數可用於通過返回 None 來隱藏邊。所以weight = lambda u, v, d: 1 if d['color']=="red" else None會找到最短的紅色路徑。

基於 https://code.activestate.com/recipes/119466/ 的 Python 食譜食譜 (119466)

如果邊權重為負數或浮點數(溢出和舍入錯誤可能導致問題),則不保證此算法有效。

例子

>>> 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]

相關用法


注:本文由純淨天空篩選整理自networkx.org大神的英文原創作品 networkx.algorithms.shortest_paths.weighted.single_source_dijkstra。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。