파이썬 그래프 그리기 (NetworkX 라이브러리)

※ matplotlib 설치 필요

python -m pip install -U matplotlib

pip install networkx

Directed Graph 일 경우, 두 노드에서 굵은 노드 있는 쪽이 방향 가르키는 쪽

ex) 아래 그래프, 노드1과 노드2는 양방향성 // 노드2와 노드3은 노드2 -> 노드3만 존재

import networkx as nx
import matplotlib.pyplot as plt

G = nx.DiGraph()

# 노드 = 원, 엣지 = 선 (add_node를 안하고 add_edge(1,2)을 해도 노드 1, 2는 자동 생성됨
G.add_nodes_from([1, 2, 3, 4])
G.add_edges_from([(1, 2), (2, 1), (2, 3), (2,4)])

# 라벨 표시안하려면 with_labels 파라미터 지우거나 False
nx.draw(G, with_labels=True)

# 이 py 파일 위치하는 곳에 사진으로 저장
plt.savefig("G.png")

plt.show()

Networkx 가 지원하는 layout 종류를 보고 싶으면 아래 링크 참조

https://frhyme.github.io/python-lib/networkx_layout/