最短路径（迪杰斯特拉算法Dijkstra）python版本实现

import numpy as np

'''
    该版本迪杰斯特拉算法 只实现原点到终点的最短路径
    对中间其他点的最短路径没有做处理
'''


def get_shortest_route(start: int, data):
    # 获取最大路径值 用于排除
    max_value = np.max(mgraph)
    # 获取列表长度
    lenth = len(mgraph)
    # while循环记录位置
    cur_index = 0
    # 默认第一条路径就为 1
    result = [start]
    while True:
        # 获取单个列表
        cur_list = mgraph[cur_index]
        tmp_dict = {}
        # 遍历列表值 及其 下标
        for index, value in enumerate(cur_list):
            if value == 0 or value == max_value:
                continue
            tmp_dict[index + 1] = value
        # 对当前字典中的下标 及 值 进行筛选过滤
        tmp_key_list = tmp_dict.keys()
        if len(tmp_key_list) > 0:
            # 如果最大路径所在的下标已经存在 则遍历完成
            if lenth in result:
                return result
            else:
                min_value = min(tmp_dict.values())
                # 获取当前字典最小值 及其 下标
                for i in tmp_dict:
                    if tmp_dict[i] == min_value:
                        result.append(i)

        cur_index += 1
        if cur_index > lenth - 1:
            break


if __name__ == "__main__":
    inf = 10086
    mgraph = [[0, 2, 5, inf, inf],
              [inf, 0, 2, 6, inf],
              [inf, inf, 0, 7, 1],
              [inf, inf, 2, 0, 4],
              [inf, inf, inf, inf, 0]]
    start = 1
    result = get_shortest_route(start, mgraph)
    print(result)