6.3

import numpy as np
import pandas as pd
import cvxpy as cp
import networkx as nx
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif']=['Times New Roman + SimSun + WFM Sans SC']
plt.rcParams['mathtext.fontset']='stix'

Times New Roman + SimSun + WFM Sans SC

simsum宋体, times new roman -*, simhei黑体, kaiti楷体,

dengxian等线, fangsong仿宋, Microsoft Yahei微软雅黑

plt.rcParams['axes.unicode_minus']=False
plt.rcParams['figure.dpi'] = 200

plt.rcParams['figure.figsize'] = [4, 3]

plt.rcParams['font.size'] = 12

plt.rcParams['xtick.direction']='in'
plt.rcParams['ytick.direction']='in'
L=[(1,2,20),(1,5,15),(2,3,20),(2,4,60),(2,5,25),
(3,4,30),(3,5,18),(4,5,35),(4,6,10),(5,6,15)]
G = nx.Graph()
G.add_weighted_edges_from(L)
T = nx.minimum_spanning_tree(G)
w = nx.get_edge_attributes(T, 'weight')
print("最小生成树的长度为:", sum(w.values()))
nodes_labels = ['v' + str(i) for i in range(1, 7)]
nodes_labels = dict(zip(range(1,7), nodes_labels))
pos = nx.spring_layout(T)
nx.draw(T, pos, with_labels=True, labels=nodes_labels, font_color='w')
nx.draw_networkx_edge_labels(T, pos, edge_labels=w)
plt.show()

posted @ 2024-10-27 23:56  世梦  阅读(11)  评论(0)    收藏  举报