グラフを可視化したい！
みなさん、競プロの問題でわけのわからないアルゴリズムを見た時、可視化したグラフでアルゴリズムを順番に精査したくなりませんか？僕はなります。そんな時、手動で描いていませんか？僕は描いていました。
もうそんなことに時間は使わない！これからはNetworkx+Matplotlibでグラフを描画しましょう。
今回使うパッケージ
Networkxは、ネットワーク分析のために作られたPythonのパッケージです。グラフの作成のみならず、分析、計算まで出来ちゃいます。ダイクストラ法やベルマンフォード法等も数行で記述できますが、時間と空間計算量を食うらしいので競プロの問題で使うのはお勧めできません。
Matplotlibは言わずと知れたグラフ描画用のパッケージです。
この記事では、Networkxでグラフを作成した後にMatplotlibでそれを描画していきます。
ライブラリのインストール
pipでインストールします
pip install networkx
pip install matplotlib
この記事のコードは、matplotlib 3.8.2, networkx 3.2.1での動作を確認しています。
単純無向グラフ
辺に方向性がないグラフです。draw()関数で矢印の引数arrowsをFalseにすることによって描くことができます。
import networkx as nx
import matplotlib.pyplot as plt
# グラフの作成
G_undirected = nx.DiGraph()
# 頂点のリスト
edges_undirected = [(0, 1), (1, 2), (1, 3), (3, 4), (1, 4)]
# グラフに辺を入れる
G_undirected.add_edges_from(edges_undirected)
# グラフの可視化
nx.draw(G_undirected, with_labels=True, arrows=False)
plt.show()
単純有向グラフ
辺に方向が存在するグラフです。無向グラフとほとんど違いなく描くことができます。双方向の辺は1本の矢印で表されます。矢印の向きは 頂点1 -> 頂点2 です。
import networkx as nx
import matplotlib.pyplot as plt
# グラフの作成
G_directed = nx.DiGraph()
# 辺のリスト
edges_directed = [(1, 2), (2, 3), (3, 4), (4, 1), (1, 3), (2, 1)]
# グラフに辺を入れる
G_directed.add_edges_from(edges_directed)
# 有向グラフの可視化
nx.draw(G_directed, with_labels=True, arrows=True)
plt.show()
重み付きグラフ
基本的には普通の有向グラフと同じですが、辺を追加するときに重みも追加することができます。
今までは頂点の位置の指定は必須ではありませんでしたが、重みを描画する際に必要なので頂点の位置を自動設定しておきます。頂点の位置はカスタマイズも可能です。点の位置を調整を参照してください。
import networkx as nx
import matplotlib.pyplot as plt
# グラフの作成
G_weighted = nx.DiGraph()
# （頂点１, 頂点2, 重み）
edges_weighted = [(1, 2, 1.5), (2, 3, 2.5), (3, 4, 3.5), (4, 1, 4.5), (1, 3, 5.0)]
# 辺を追加
G_weighted.add_weighted_edges_from(edges_weighted)
# 頂点の位置の自動設定
pos = nx.spring_layout(G_weighted)
# グラフの可視化
nx.draw(G_weighted, pos, with_labels=True, arrows=True)
# 重みのラベルを追加
edge_labels = nx.get_edge_attributes(G_weighted, 'weight')
nx.draw_networkx_edge_labels(G_weighted, pos, edge_labels=edge_labels)
plt.show()
完全グラフ
全ての頂点が互いに辺を持つグラフです。
import networkx as nx
import matplotlib.pyplot as plt
# 頂点が6つある完全グラフ
G_complete = nx.complete_graph(6)
# グラフの可視化
nx.draw(G_complete, with_labels=True)
plt.show()
点の位置を調整
draw()関数の中で、posという点の位置を指定する引数があります。辞書型で座標を渡すことができます。
import networkx as nx
import matplotlib.pyplot as plt
# 有向グラフの作成
G_directed = nx.DiGraph()
edges_directed = [(1, 2), (2, 3), (3, 4), (4, 1), (1, 3)]
G_directed.add_edges_from(edges_directed)
# 頂点の位置を設定
pos = {1: (0, 0), 2: (0.5, 0.5), 3: (1, 0), 4: (0.5, -0.5)}
# 座標を渡して可視化
nx.draw(G_directed, pos, with_labels=True, arrows=True)
plt.show()
辺の追加、削除
# 辺の追加
G.add_edge(0, 1)
G.add_edges_from([(1, 2), (2, 3)])
# 重み付きグラフへの辺の追加
weighted_G.add_edge(1, 2, weight=3.5)
# 辺の削除
G.remove_edge(0, 1)
G.remove_edges_from([(1, 2), (2, 3)])
終わりに
とりあえずこれだけ出来たらグラフの描画はできそう…？他にもやりたいことがあれば追記していきます。