元进Python(2)---networkx和matplotlib联手制作网络IP拓扑图

import networkx as nx
import matplotlib.pyplot as plt
#引入两个模块，如果没有，需要在pip install。

MyNode=[10,5,1,2,6,11]
MyWeb=[]
MyPos={}
MyLabel={}
#定义节点、链路、节点位置和标签。

i=0
while i<len(MyNode)-1:
    MyWeb.append((MyNode[i],MyNode[i+1]))
    i+=1
#生成相连链路数列，供后面nx生成图使用

i=0
while i<len(MyNode):
    MyPos.update({MyNode[i]:[i%3,i//3]})
    i+=1
#生成节点位置字典

i=0
tag=[]
while i<len(MyNode)-1:
    if MyNode[i]<MyNode[i+1]:
        tag.append('100.'+str(MyNode[i])+'.'+str(MyNode[i+1])+'.1/2')
    else:
        tag.append('100.'+str(MyNode[i+1])+'.'+str(MyNode[i])+'.2/1')
    i+=1
#生成节点相连链路两端的IP地址列表

for x in MyWeb:
    MyLabel.update({x:tag[MyWeb.index(x)]})
#让IP地址列表变成节点相连链路的标签

G=nx.Graph()
G.add_nodes_from(MyNode)
G.add_edges_from(MyWeb)
#添加节点和链路

nx.draw(G,pos=MyPos,with_labels=True,font_size='13',font_color='yellow',edge_color='r',
        node_shape='o',node_color='green', node_size=1700)
#生成节点和链路图

nx.draw_networkx_edge_labels(G,pos=MyPos,
edge_labels=MyLabel,font_color='red',font_family='Arial',font_size='12')
#生成节点相连链路标签图

plt.show()
#plt显示图形