欢迎使用CSDN-markdown编辑器

【python】pygraphviz生成二叉树图

pytgraphviz是graphviz工具python包装库。在本质上还是使用dot语言，实现画图功能。其最主要的功能点，其实只是快速的生成.dot文件，绘图的功能需要dot的支持。

---

绘图

先上代码：

#画图
def draw_plshu():
    r_tulp = getAll()
    r_map = tulpToMap(r_tulp)
    A = pgv.AGraph(directed=True,strict=True)
    for item in r_tulp:
        l_next = 2*item[0]
        r_next = 2*item[0]+1
        node = "("+str(item[0])+")"+item[1]

        if r_map.has_key(l_next):
            l_node = "("+str(r_map[l_next][0])+")"+r_map[l_next][1]
            A.add_edge(node,l_node)
        if r_map.has_key(r_next):
            r_node = "("+str(r_map[r_next][0])+")"+r_map[r_next][1]
            A.add_edge(node,r_node)

        A.graph_attr['epsilon']='0.001'
        A.write('pl003.dot')
        A.layout('dot')
        A.draw('pl003.svg')

AGraph对象是python包装的graph对象，使用此对象完成绘图。

方法or属性

add_*

add系列方法，程序结构中增加各个dot元素。

    A.add_edge(node,r_node)

graph_attr

graph_attr属性设置dot的graph属性，如dot中的：

strict digraph {
    graph [bb="0,0,52843,684",
        epsilon=0.001
    ];
}

write

write()方法写出dot文件

draw

draw()方法调用dot画图

---

注意使用画图时，最好使用svg格式的矢量图。使用png格式的位图，遇到数据量大时，会导致图片画不出来。