matlab有向图的绘制,【Python学习系列二十五】数据结构-有向图绘制

1、场景：从文件中读取节点、有向边，然后绘制。

2、参考代码：

# -*- coding: utf-8 -*-

import networkx as nx

import matplotlib.pyplot as plt

#读取文件，获取节点和边

f = open("D:\\tmp\\gy_contest_link_top.txt", "r")

nodelist=[]

edgelist=[]

while True:

line = f.readline()

if line:

pass # do something here

line=line.strip()

node=line.split(';')[0]#获取图节点

nodelist.append(node)

in_nodes=line.split(';')[1].split('#')#获取图边，该节点是终点

for ins in range( len(in_nodes) ) :

if in_nodes[ins].strip() !='':

in_edge=(in_nodes[ins],node)

if in_edge not in edgelist:

edgelist.append(in_edge)

out_nodes=line.split(';')[2].split('#')#获取图边，该节点是起点

for ins in range( len(out_nodes) ) :

if out_nodes[ins].strip() !='':

out_edge=(node,out_nodes[ins])

if out_edge not in edgelist:

edgelist.append(out_edge)

else:

break

f.close()

del nodelist[0] #删除表头生成的节点

del edgelist[0]

del edgelist[0] #删除表头生成的边

#print len(nodelist) #图节点

#print len(edgelist) #边数

#有向图绘制

G=nx.DiGraph()

G.add_nodes_from(nodelist)

G.add_edges_from(edgelist)

nx.draw_networkx(G, pos=None, arrows=True, with_labels=True)

#plt.savefig('D:\\tmp\\it.png')

plt.show()

效果图：

笔者没有对节点名做简易处理，所以看起来有点乱。