networkx绘图简易教学

参考networkx 的图形简单操作

如下代码会添加一些节点和边，并标注颜色。注意到颜色的涂色顺序与节点被添加的顺序一致。

import networkx as nx
from matplotlib import pyplot as plt

def graph_test01():
    # 图的生长
    g = nx.Graph()
    # 添加节点
    g.add_node(88)                       # 添加一个节点
    g.add_nodes_from([11, 26, 100])        # 添加一个列表节点
    g.add_nodes_from([45, 3, 15])
    # 添加边
    g.add_edges_from([(11, 26), (26, 100)])  # 添加一条多条连边
    g.add_edges_from([(45, 3)])
    g.add_edges_from([(3, 15)])

    color_map = ["y"]
    color_map.extend(["red"] * 3)
    color_map.extend(["green"] * 3)

    nx.draw(g, with_labels=True, node_color=color_map)        # 绘图,并显示节点序号
    plt.show()                          # 显示
    g.clear()                           # 删除图中的所有连边和节点


if __name__ == '__main__':
    graph_test01()