PythonからGraphvizを使う

環境はMacでPython3.5です。

Graphvizというdot言語を用いてグラフを作成するツールがあるので使ってみた。

インストール

# コマンドから使用するときは
brew install graphviz
# Pythonから呼び出したいときは
pip install graphviz

コード

from graphviz import *

G = Digraph(format="png") # 有向グラフ初期化
G.attr('node', shape='circle') # ノードの形

G.node("my","俺") # ノード作成

N =  1  # ノード数
for i in range(N):
    G.node("boss"+str(i),"ボス") 
    G.edge("boss"+str(i), "my" ) # edge(元, 先) で線をつなぐ

print(G)
G.render('my') # 画像作成

pngファイルが生成される。

f:id:carumisu:20160623111815p:plain