图邻接矩阵可视化解析
短信预约 -IT技能 免费直播动态提醒
使用工具
#导入模块
import networkx as nx
import numpy as np
import matplotlib.pyplot as plt
import scipy.sparse as sp
准备数据
# 邻接矩阵
Matrix = np.array(
[
[0, 1, 1, 1, 1, 1, 0, 0], # a
[0, 0, 1, 0, 1, 0, 0, 0], # b
[0, 0, 0, 1, 0, 0, 0, 0], # c
[0, 0, 0, 0, 1, 0, 0, 0], # d
[0, 0, 0, 0, 0, 1, 0, 0], # e
[0, 0, 1, 0, 0, 0, 1, 1], # f
[0, 0, 0, 0, 0, 1, 0, 1], # g
[0, 0, 0, 0, 0, 1, 1, 0] # h
]
)
转化临界矩阵
def get_matrix_triad(coo_matrix , data=False):
'''
获取矩阵的元组表示 (row,col)
data 为 True 时 (row,col,data)
:dependent scipy
:param coo_matrix: 三元组表示的稀疏矩阵 类型可以为 numpy.ndarray
:param data: 是否需要 data值
:return:
list
'''
# 检查类型
if not sp.isspmatrix_coo(coo_matrix):
# 转化为三元组表示的稀疏矩阵
coo_matrix = sp.coo_matrix(coo_matrix)
# nx3的矩阵 列分别为 矩阵行,矩阵列及对应的矩阵值
temp = np.vstack((coo_matrix.row , coo_matrix.col , coo_matrix.data)).transpose()
return temp.tolist()
测试
edags = get_matrix_triad(Matrix)
-->
[[0.0, 0.0, 1.0],
[0.0, 1.0, 1.0],
[0.0, 2.0, 1.0],
[0.0, 3.0, 1.0],
[0.0, 4.0, 1.0],
[0.0, 5.0, 1.0],
[1.0, 1.0, 1.0],
[1.0, 2.0, 1.0],
[1.0, 4.0, 1.0],
[2.0, 2.0, 1.0],
[2.0, 3.0, 1.0],
[3.0, 3.0, 1.0],
[3.0, 4.0, 1.0],
[4.0, 4.0, 1.0],
[4.0, 5.0, 1.0],
[5.0, 2.0, 1.0],
[5.0, 5.0, 1.0],
[5.0, 6.0, 1.0],
[5.0, 7.0, 1.0],
[6.0, 5.0, 1.0],
[6.0, 6.0, 1.0],
[6.0, 7.0, 1.0],
[7.0, 5.0, 1.0],
[7.0, 6.0, 1.0],
[7.0, 7.0, 1.0]]
创建图
# 创建一个没有边,没有节点的空图Graph
G = nx.Graph()
添加节点
按照节点的个数添加节点
H = nx.path_graph(Matrix.shape[0])
G.add_nodes_from(H)
添加边
G.add_edges_from(edags) #添加边
# 若数据含有权重,及 get_matrix_triad() 中 data = True ,则使用
G.add_weighted_edges_from(edags)
绘图
colors = np.arange(Matrix.shape[0])
nx.draw(G,pos=nx.spring_layout(G),node_color=colors)
plt.show()
扩展
美化图
合理使用**draw_networkx ()**中的参数,来美化图
draw_networkx()
https://networkx.github.io/documentation/stable/reference/generated/networkx.drawing.nx_pylab.draw_networkx.html#networkx.drawing.nx_pylab.draw_networkx
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持编程网。
免责声明:
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341