pythonにおけるGraphvizの簡単なまとめ(macの場合のみ解説)


Graphvizのインストール手順

1. homebrewでgraphvizをmacにインストール

brew install graphviz

2. pipでgraphvizをインストール

pip install graphviz

ノードを生成

コード

from graphviz import Digraph

# pngとして出力
dg = Digraph(format='png')

dg.node('1') # 1というラベルがついたノードを設定

dg.node('2') # 2というラベルがついたノードを設定

dg.render('./test/dgraph1') # テストフォルダにdgraphという名前で保存(拡張子は書かない)

出力結果

エッジを生成

コード

from graphviz import Digraph

dg = Digraph(format='png')

dg.node('1') # 1というラベルがついたノードを設定
dg.node('2') # 2というラベルがついたノードを設定

dg.edge('1','2') # ノード1 からノード2に向かって矢印を描く

dg.render('./test/dgraph2') # テストフォルダにdgraphという名前で保存

出力結果

ノードの形(shape)を変更

形の種類はこちらを参照

コード

from graphviz import Digraph

dg = Digraph(format='png')

# デフォルトを○に設定
dg.attr("node", shape="circle")

dg.node('1') # 1というラベルがついたノードを設定(○)
dg.node('2') # 1というラベルがついたノードを設定(○)
dg.node('3',shape='box') # 特定のノードを□に変更


dg.render('./test/dgraph') # テストフォルダにdgraphという名前で保存

出力結果

参考文献

Mac : Pythonでgraphvizを使う方法【インストールから動作確認まで】
PythonとGraphvizでノードのフォントサイズを指定する
Node Shapes