[Python] pltとnetworkxでグラフを可視化してみる

Python

2020.01.03

pythonでnetworkxというライブラリを発見し、便利だと思ったので

投稿しようと思います。

networkxは外部ライブラリなので、インストールしてください

pip install networkx

と実行すればインストールできると思います。

グラフを可視化してみよう!!!

matplotlib.pyplotを使うのでimportしてください。

import networkx as nx
import matplotlib.pyplot as plt

V,E=map(int,input().split()) #ノードの数、辺の数

G=nx.DiGraph() #有向グラフを生成

for i in range(E):
    a,b,w=map(int,input().split()) #始点、終点、重さ
    G.add_edge(str(a),str(b),weight=w)

nx.draw_networkx(G)
plt.show()

"""
4 4
0 1 3
1 2 4
2 3 4
3 0 2
"""