pydot包提供了使用dot语言的python接口,可以方便地创建各种有向图和无向图。
官方网站:https://code.google.com/p/pydot/ (目前pydot只支持到graphviz-2.26.3)
安装需要:
1.pyparsing http://pyparsing.wikispaces.com/ (我下载的pyparsing-1.5.7.win32-py2.7.exe)
2.Graphviz http://www.graphviz.org/pub/graphviz/stable/windows/ (我下载的graphviz-2.26.3.msi)
所有安装包下载点这里
我的安装环境为Windows 7,python-2.7(系统环境变量path中已添加python路径)
安装:
1. 将pyparsing和Graphviz直接单击安装
2. 解压pydot包,在终端进入目录,然后输入:python setup.py install
使用:
# -*- coding: cp936 -*-
import pydot
g = pydot.Dot('mygraph',g_type='dig') #创建有向图
node = pydot.Node(1)#创建节点'1',与pydot.Node('1')等价
g.add_node(node) #添加节点
print node.get_name() #输出为字符串1
node = pydot.Node(2,label='bbb') #如果设置label属性,那么画图时节点显示为'bbb'
g.add_node(node)
print node.get_name() #输出为字符串2
node = pydot.Node(3,label='ccc')
g.add_node(node)
e = pydot.Edge('1','2') #创建边1->2
g.add_edge(e) #添加边
e = pydot.Edge('1','3') #创建边1->3
g.add_edge(e) #添加边
print g.to_string() # 打印整个图
##digraph mygraph {
##g_type=dig;
##1;
##2 [label=bbb];
##3 [label=ccc];
##1 -> 2;
##1 -> 3;
##}
g.write_jpg("mygraph.jpg") #保存图形到文件
在python shell中可以输入help(pydot.Dot), hep(pydot.Node), help(pydot.Edge) 查看更多函数的用法。
其他参考: