因为要用到一些树的图形,所以搜索到了 PyGraphviz 这个绘图工具。PyGraphviz 是对 Graphviz 的封装,提供了 Python 接口的调用。Graphviz 是一个开源软件包,提供了对图、点、边的简易操作,所以封装后的 PyGraphviz 可以很容易用来绘制想要的图形。
下面是 Windows 系统下具体安装过程:
pip install pygraphviz-1.3.1-cp34-none-win_amd64.whl
import pygraphviz as pyg
g=pyg.AGraph() #建立图
g.add_node('A') #建立点
g.add_edge('A','B') #建立边
g.add_edge('A','C') #建立边
g.layout(prog='dot') #绘图类型
g.draw('pyg1.png') #绘制
绘制结果:
pyg1.png
PyGraphviz 对于图、点和边的设置,如颜色、样式、形状等属性,分别提供有 graph_attr、node_attr 和 edge_attr 属性设置函数。
以上步骤中请注意 Graphviz 安装完成后,记得添加安装目录下的 bin 目录到系统 Path 中,不然会提示如下两种错误:
ValueError: No prog dot in path.
ValueError: Program dot not found in path.