파이썬 networkx weight 추가 & 그리기

※ matplotlib 설치 필요

import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()

# 원하는 파라미터 이름 = 실수 (**kwds)
G.add_edge(1, 2, weight=5)
G.add_edge(2, 3, weight=3)
G.add_edge(3, 4, weight=2)
G.add_edge(2, 4, weight=1)
G.add_edge(1, 4, weight=0.5)

# 아래는 그래프로 그리는 법
pos=nx.spring_layout(G)
nx.draw(G, pos=pos, with_labels=True)

# edge 라벨은 weight 값을 가져온다.
labels = nx.get_edge_attributes(G,'weight')

nx.draw_networkx_edge_labels(G,pos,edge_labels=labels)

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