用python狄克斯特拉算法求最短路径

用python狄克斯特拉算法求最短路径

#用python狄克斯特拉算法求最短路径


import networkx as nx
from matplotlib import pylab
import numpy as np

# 自定义网络
row = np.array([0, 0, 1, 1, 2, 2,3,4])#起点初始位置
col = np.array([1, 2, 3, 4, 3, 4,5,5])#起点的对应位置
value = np.array([5, 0, 15, 20, 30, 35,20,10])#起点到对应位置路径权重
atarget=max(col)
print('生成一个空的有向图')
G = nx.DiGraph()
print('为这个网络添加节点...')
for i in range(0, np.size(col) + 1):
      G.add_node(i)
print('在网络中添加带权中的边...')
for i in range(np.size(row)):
    G.add_weighted_edges_from([(row[i], col[i], value[i])])
print('给网路设置布局...')
pos = nx.shell_layout(G)
print('画出网络图像：')
nx.draw(G,pos,with_labels=True, node_color='white', edge_color='red', node_size=400, alpha=0.5 )
pylab.title('Self_Define Net',fontsize=15)
pylab.show()
'''
Shortest Path with dijkstra_path
'''
print('dijkstra方法寻找最短路径：')
path = nx.dijkstra_path(G, source=0, target=atarget)
print('节点0到'+str(atarget)+'的路径：', path)
print('dijkstra方法寻找最短距离：')
distance = nx.dijkstra_path_length(G, source=0, target=atarget)
print('节点0到'+str(atarget)+'的路径：', distance)

上述代码运行结果如下
dijkstra方法寻找最短路径：
节点0到5的路径： [0, 1, 4, 5]
dijkstra方法寻找最短距离：
节点0到5的路径： 35