python实现Floyd算法

废话不多说,直接上代码!
一、导入邻接矩阵

import xlrd
import sys

def matrix(address):                           #读取excel生成邻接矩阵
    wb = xlrd.open_workbook(address)
    sheet1 = wb.sheet_by_name('邻接矩阵_距离')
    L = []
    for i in range(1,13):
        a = sheet1.row_values(i)
        a.remove(a[0])
        L.append([int(x) for x in a])
    # print(L)
    return L

二、实现Floyd算法

from pylab import *
import matrix


def floyd(d):
    D=d
    lengthD = len(D)                    #邻接矩阵大小
    p = list(range(lengthD))
    P = []
    for i in range(lengthD):
        P.append(p)
    P = array(P)
    for k in range(lengthD):
        for i in range(lengthD):
            for j in range(lengthD):
                if(D[i][j] >D[i][k]+D[j][k]):         #两个顶点直接较小的间接路径替换较大的直接路径
                    P[i][j] = P[i][k]                 #记录新路径的前驱
                    D[i][j] = D[i][k]+D[j][k]
    print('各个顶点的最短路径:')
    for i in range(lengthD):
        for j in range(i+1,lengthD):
            print('v%d' % (i+1) + '--' + 'v%d' % (j+1) + '\t' + 'dist_min:' + '\t' + str(D[i][j]) + '\t' + 'path:'+'v%d'%(i+1),end='' )
            temp=P[i][j]
            while (temp!=j):
                print('--'+'v%d'%(temp+1),end='')
                temp=P[temp][j]
            print('--'+'v%d'%(j+1))
    print('P矩阵:')
    print(P)
    print('D矩阵:')
    for i in D:
        print(i)
    return D


if __name__ == '__main__':
    chararray = ['A','B','C','D','E','F','G','H','I','J','K','L']
    L=[]
    for i in range(1,len(chararray)+1):
        L.append('v%d'%i)
    Dict=dict(zip(L,chararray))
    print('各点对应关系:')
    print(Dict)
    address = 'E:/py/math_b/b题数据和代码/b题基础数据.xlsx'
    data=matrix.matrix(address)
    d=matrix.tran_m(data)
    floyd(d)

三、仿真结果
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/heavenmark/article/details/82794565