graphviz绘制图和树,python(一)

x33g5p2x  于2021-11-11 转载在 Python  
字(1.0k)|赞(0)|评价(0)|浏览(381)

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()

输出如同:

相关文章