绘制网络分面图—matplotlib，networkx

来源：互联网发布：石化盈科网络大学编辑：程序博客网时间：2024/06/10 22:26

使用python的matplotlib与networkx绘制网络分面图

数据格式：

左边两列是网络的ID与name 再右边变是网络的连边了。比如第一行网络的边为（1,2）（3,4）（2,3）（4,5）

代码：

# coding: utf-8import networkx as nximport matplotlib.pyplot as pltimport csvdef LoadData():    UnitedVisibilityGraphDictData=csv.reader(file('UnitedVisibilityGraphDict.csv', 'rb'))    UnitedVisibilityGraphDict={}    count=0  #don't read first line    for data in UnitedVisibilityGraphDictData:        edges=[]        if count!=0:             count2=2             for i in range(len(data)/2-1):                if len(data[count2])!=0:                   medium=(int(data[count2]),int(data[count2+1]))                   edges.append(medium)                   count2+=2             UnitedVisibilityGraphDict[data[0]]=edges        count+=1    #print UnitedVisibilityGraphDict['1']  #test code    #print UnitedVisibilityGraphDict['4']    return UnitedVisibilityGraphDictdef DrawVisibilityGraphDict(UnitedVisibilityGraphDict):    graphList=[]    for i in range(1,len(UnitedVisibilityGraphDict)+1):         G=nx.Graph()         G.add_edges_from(UnitedVisibilityGraphDict[str(i)])         graphList.append((i,G))  #i is int type ,G is class type    fig=plt.figure(figsize=(20,16))    #plt.clf()#clear  canvas    for i in range(1,26):          ax=fig.add_subplot(5,5,i)          ax.set_xticks([])          ax.set_yticks([])          ax.set_ylim([-0.2,1.2])          plt.title('Visibility Graph State %d'%(graphList[i-1][0]))          #nx.draw(graphList[i-1][1],pos,node_size=150,with_labels=True)  #this method can draw the picture without frame          pos=nx.spring_layout(graphList[i-1][1],iterations=30)          nx.draw_networkx_nodes(graphList[i-1][1],pos,node_size=400)          nx.draw_networkx_edges(graphList[i-1][1],pos,width=2)          nx.draw_networkx_labels(graphList[i-1][1],pos,font_size=20,font_family='sans-serif')    plt.savefig("outputResult/VisibilityGraphDict2.png")    plt.show()if __name__=="__main__":    UnitedVisibilityGraphDict=LoadData()    DrawVisibilityGraphDict(UnitedVisibilityGraphDict)

结果：

0 0