使用networkx画图,关键是掌握draw_networkx_edges、draw_networkx_nodes和draw_networkx_labels三个方法的使用,设置好位置列表,节点列表,标签列表,理论上就可以绘制出任意的图了。详情参见:https://networkx.github.io,示例如下。
效果:
代码: options = { 'node_color': 'red', 'node_size': 300, 'with_labels': True, } G = createGraph(20, 5, 0.1) g, rings = getRelaxTree(G) # 画生成图 plt.sca(plt.subplot(121)) nx.draw_shell(G, **options) # 画松弛树 plt.sca(plt.subplot(122)) pos = nx.shell_layout(g) li = list(g.nodes) nx.draw_networkx_nodes(g, pos, nodelist=li) nx.draw_networkx_edges(g, pos) # 将R节点用不同颜色标记 R