pip install graphviz
图片保存形式:pdf或png,在初始化时指定
# -*- coding: utf-8 -*-
# @file : graphviz_demo.py
# @date : 2018-07-03
# @author : peng shiyu
from graphviz import digraph
dot = digraph(comment='the test table', format="png")
# 添加圆点a,a的标签是dot a
dot.node('a', 'dot a')
# 添加圆点 b, b的标签是dot b
dot.node('b', 'dot b')
# dot.view()
# 添加圆点 c, c的标签是dot c
dot.node('c', 'dot c')
# dot.view()
# 创建一堆边,即连接ab的两条边,连接ac的一条边。
dot.edges(['ab', 'ac', 'ab'])
# dot.view()
# 在创建两圆点之间创建一条边
dot.edge('b', 'c', 'test')
# dot.view()
# 获取dot source源码的字符串形式
# print(dot.source)
# // the test table
# digraph {
# a [label="dot a"]
# b [label="dot b"]
# c [label="dot c"]
# a -> b
# a -> c
# a -> b
# b -> c [label=test]
# }
# 保存source到文件,并提供graphviz引擎
dot.save('test-table.gv') # 保存
dot.render('test-table.gv')
# dot.view() # 显示
# 从保存的文件读取并显示
from graphviz import source
s = source.from_file('test-table.gv')
print(s.source) # 打印代码
# s.view() # 显示
参考:
python中graphviz库的使用