python---通过networkx使图着色结果可视化

假定邻接矩阵为:
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、使图中结点显示标号,并按特定颜色显示

  • 4
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值