习题6.5
习题6.5代码
import numpy as np
import cvxpy as cp
import networkx as nx
n = 6
A=np.zeros((n,n))
A[0,[1,2]]=[2,7]
A[1,2:5]=[4,6,8]
A[2,[3,4]]=[1,3]
A[3,[4,5]]=[1,6]
A[4,5]=3
G = nx.Graph(A)
dis = nx.floyd_warshall_numpy(G)
dis_max = np.max(dis, axis=1)
hosp_loc = np.argmin(dis_max) + 1
hosp_loc
stu = np.array([50, 40, 60, 20, 70, 90])
stu_dis = stu@dis
scho_loc = np.argmin(stu_dis) + 1
print(scho_loc)
