利用NetworkX库进行图论绘图

最新推荐文章于 2024-12-14 22:58:16 发布

原创

最新推荐文章于 2024-12-14 22:58:16 发布 · 636 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍如何利用Python的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)+str(k))
            G.add_edge(str(j),str(j)+str(k),weight=layer-i+1)
        print rootdict[i]            
    

#sys.ex