因為要用到一些樹的圖形遗锣,所以搜索到了 PyGraphviz 這個繪圖工具。PyGraphviz 是對 Graphviz 的封裝稀拐,提供了 Python 接口的調(diào)用拐邪。Graphviz 是一個開源軟件包,提供了對圖萝快、點锻霎、邊的簡易操作,所以封裝后的 PyGraphviz 可以很容易用來繪制想要的圖形揪漩。
下面是 Windows 系統(tǒng)下具體安裝過程:
1. 安裝 python
- 選擇 python 版本并下載旋恼,這里示例使用的是 python 3.4.4,msi 格式的文件奄容。 python下載鏈接
- 安裝 python 并添加安裝目錄到系統(tǒng)變量中的 Path 中冰更,同時添加 Scripts 目錄到 Path 中。以便于后續(xù)使用 python.exe 和 pip.exe 兩個執(zhí)行文件昂勒。
2. 安裝 Graphviz
- 下載 Graphviz 安裝包并安裝蜀细,這里示例使用的是 Graphviz2.38,Graphviz下載鏈接
- 安裝后添加安裝目錄下 bin 目錄到系統(tǒng) Path 中戈盈,以便后續(xù)使用 dot.exe 文件
3. 安裝 PyGraphviz
- 下載 PyGraphviz 并安裝奠衔,這里示例使用的是 PyGraphviz 1.3.1-cp34 版本谆刨,其中 cp34 表示 python 版本是 3.4,PyGraphviz下載鏈接
- pip 安裝下載后的 whl 文件
pip install pygraphviz-1.3.1-cp34-none-win_amd64.whl
4. PyGraphviz 使用
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') #繪制
繪制結(jié)果:
pyg1.png
PyGraphviz 對于圖归斤、點和邊的設(shè)置痊夭,如顏色、樣式脏里、形狀等屬性她我,分別提供有 graph_attr、node_attr 和 edge_attr 屬性設(shè)置函數(shù)迫横。
以上步驟中請注意 Graphviz 安裝完成后番舆,記得添加安裝目錄下的 bin 目錄到系統(tǒng) Path 中,不然會提示如下兩種錯誤:
ValueError: No prog dot in path.
ValueError: Program dot not found in path.