graphviz绘制图和树，python

Posted 2021-12-04 zhangphil

1、首先安装：

pip install graphviz

2、然后到graphviz官网（Download | Graphviz）下载Windows的exe安装文件：

安装graphviz文件时候，注意勾选把它添加到系统路径中。完成以上两步后就可以用graphviz绘制树或图。如果明明已经安装graphviz的exe文件且已经添加到系统路径，但是运行提示报错找不到可执行文件，那么，重启电脑。

3、一个例子：

from graphviz import Digraph


def my_grap():
    dot = Digraph(name="pic", comment="测试", format="png")

    # 绘制方向。默认自顶向下TB，BT自底向上，LR:左到右
    dot.attr('graph', rankdir='TB')

    # 定义图中的节点
    dot.node(name='a', label='A', color='red')
    dot.node(name='b', label='B', color='green')
    dot.node(name='c', label='C', color='yellow')
    dot.node(name='d', label='D', color='blue')

    # 定义节点与节点之间的边线
    # tail_name开始，head_name结束
    dot.edge(tail_name='a', head_name='b', label='3', color='red')
    dot.edge(tail_name='b', head_name='c', label='2', color='blue', dir='none')  # dir=none,边线不带箭头
    dot.edge(tail_name='a', head_name='d', label='4', color='yellow', dir='none')
    dot.edge(tail_name='b', head_name='d', label='5', color='green', taillabel='b->d')  # 边线标签文本

    # 存放在当前目录
    dot.render(filename='my_pic',
               directory='.',  # 当前目录
               view=True)


if __name__ == '__main__':
    my_grap()

输出如同：