floyd

nv=4
INF=999
def floyd(G):
dist=list(map(lambda i:list(map(lambda j:j,i)),G))
for k in range(nv):
for i in range(nv):
for j in range(nv):
dist[i][j]=min(dist[i][j],dist[i][k]+dist[k][j])
print_sol(dist)

def print_sol(dist):
for i in range(nv):
for j in range(nv):
if(dist[i][j]==INF):
print("INF",end=" ")
else:
print(dist[i][j],end=" ")
print(" ")

G=[[0,5,INF,INF],
[50,0,15,5],
[30,INF,0,15],
[15,INF,5,0]]
floyd(G)