文件中共定义了三个类:Dot,Graph,Digraph,后两者都是继承Dot. 一般用户绘制决策树、神经网络图等


API

  1. "类库graphviz的使用, 用于生成决策树、流程图效果拔群"
  2. import graphviz
  3. # node_attr更改节点的样式
  4. dot = graphviz.Digraph(comment='The Round Table', node_attr={"shape": "plaintext"})
  5. # 设置所有节点的样式
  6. dot.attr("node", shape="rarrow")
  7. dot.node(name='a',color='red')
  8. dot.node(name='b',color='blue')
  9. dot.node(name='c',color='blue')
  10. dot.node(name='d',color='blue')
  11. dot.edge('a','b',color='green', label="sd")
  12. dot.edges(["ac", "bd"])
  13. print(dot.source)
  14. # 生成的pdf缓存png格式
  15. dot.format = 'png'
  16. dot.render('./gv/test-table.gv', view=True)
  17. # Source从一个dot源码中进行渲染
  18. graph = graphviz.Source(dot)
  19. graph.view()