假定邻接矩阵为:
0 1 1 1 0 0 1 0
1 0 1 1 1 0 0 0
1 1 0 0 1 1 1 0
1 1 0 0 1 0 1 0
0 1 1 1 0 1 1 1
0 0 1 0 1 0 0 1
1 0 1 1 1 0 0 1
0 0 0 0 1 1 1 0
根据图着色算法已求得着色方案:
1 2 3 3 1 2 2 3
接下来可视化图着色结果
import networkx as nx #导入NetworkX包,为了少打几个字母,将其重命名为nx
import matplotlib.pyplot as plt #导入绘图包matplotlib
file_read = open('tt.txt', 'r')
file_color = open('c.txt', 'r')
try:
#读入二维邻接矩阵
matrix = [[0 for x in range(8)] for y in range(8)] #8*8的零矩阵
i = 0
for line in file_read: #读入每一行
matrix[i] = line.split(' ') #将每一行按空格分割
i = i + 1
#创建图
G = nx.Graph() #建立一个空的无向图
v = range(1, 8) #一维行向量,从1到8递增
G.add_nodes_from(v) #从v中添加结点,相当于顶点编号为1到8
line = file_color.read() #读取颜色向量
colors = (line.split(' ')) #颜色向量
for i in range(len(colors)):
colors[i] = int(colors[i]) #将字符转为数字
for x in range(0, len(matrix)): #添加边
for y in range(0, len(matrix)):
if matrix[x][y] == '1':
G.add_edge(x, y)
#绘制网络图G,带标签, 用指定颜色给结点上色
nx.draw(G, with_labels=True, node_color=colors)
plt.show() #输出方式: 在窗口中显示这幅图像
finally:
file_read.close()
心得:
通过该例学会了python中以下知识点
1、二维数组创建
2、遍历二维数组
3、文件按行读取
4、使用split对字符串进行分割
5、将字符转为数字
巩固以下python中知识点
1、图的创建
2、添加结点、边
3、使图中结点显示标号,并按特定颜色显示