python—networkx:根据图的权重画图

前端之家收集整理的这篇文章主要介绍了python—networkx:根据图的权重画图前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

首先输入边和边的权重,随后画出节点位置,根据权重大小划分实边和虚边

  1. import networkx as nx

  2. G=nx.Graph()

  3. 添加带权边

  4. G.add_edge('a','b',weight=0.6)

  5. G.add_edge('a','c',weight=0.2)

  6. G.add_edge('c','d',weight=0.1)

  7. G.add_edge('c','e',weight=0.7)

  8. G.add_edge('c','f',weight=0.9)

  9. G.add_edge('a',weight=0.3)

  10. 按权重划分为重权值得边和轻权值的边

  11. elarge=[(u,v) for (u,v,d) in G.edges(data=True) if d['weight'] >0.5]

  12. esmall=[(u,d) in G.edges(data=True) if d['weight'] <=0.5]

  13. 节点位置

  14. pos=nx.spring_layout(G) # positions for all nodes

  15. 首先画出节点位置

  16. nodes

  17. nx.draw_networkx_nodes(G,pos,node_size=700)

  18. 根据权重,实线为权值大的边,虚线为权值小的边

  19. edges

  20. nx.draw_networkx_edges(G,edgelist=elarge,width=6)

  21. nx.draw_networkx_edges(G,edgelist=esmall,width=6,alpha=0.5,edge_color='b',style='dashed')

  22. labels标签定义

  23. nx.draw_networkx_labels(G,font_size=20,font_family='sans-serif')

  24. plt.axis('off')

  25. plt.savefig("weighted_graph.png") # save as png

  26. plt.show() # display

  1.  

猜你在找的Python相关文章