利用NetworkX库进行图论绘图

#-*- coding: utf-8 -*- 
import matplotlib.pyplot as plt
import networkx as nx                            #导入NetworkX包，为了少打几个字母，将其重命名为nx
import random
import sys


G = nx.Graph()                                        #建立一个空的无向图G
'''
G.add_node(1)                                        #添加一个节点1
G.add_edge(2,3)                                     #添加一条边2-3（隐含着添加了两个节点2、3）
G.add_edge(3,2)                                     #对于无向图，边3-2与边2-3被认为是一条边
'''

maxsub=4
layer=4

rootdict={}

rootdict[0]='1'
G.add_node('1')                                     

for i in range(1,layer):
    rootdict[i]=[]
    for j in rootdict[i-1]:
        tmprandom = random.randint(2, maxsub)
        print j,' tmprandom:',tmprandom
        for k in range(1,tmprandom+1):
            rootdict[i].append(str(j)+str(k))
            G.add_node(str(j)