pydot 基础模板

import pydot


def test_pydot():
    graph = pydot.Dot("test_graph", graph_type="graph", bgcolor="yellow")

    # add nodes

    my_node = pydot.Node("a", label="foo")
    graph.add_node(my_node)

    graph.add_node(pydot.Node("b", shape="circle"))

    # add edges
    my_edge = pydot.Edge("a", "b", color="blue")
    graph.add_edge(my_edge)

    graph.add_edge(pydot.Edge("b", "c", color="blue"))

    graph.write_svg("test.svg")
    import os

    assert os.path.exists("test.svg")

基础逻辑为，先建立graph，再建立nodes，然后建立连接两个nodes的edge。

另外，提一下，在ubuntu下安装它的依赖graphviz的时候，只需要借助conda，然后：

conda install graphviz

就好了。不像windows下挺麻烦。强推ubuntu远程+vscode(windows)本地。