NetworkX 그래프에서 엣지 레이블의 오프셋(위치)을 조정하려면 draw_networkx_edge_labels() 함수의 label_pos 매개변수를 활용하면 됩니다. 이 값은 레이블이 엣지의 어느 지점에 놓일지 결정하며, 0.0은 시작 노드 쪽, 1.0은 도착 노드 쪽, 기본값인 0.5는 엣지의 정중앙을 의미합니다.
구현 단계
- 그림(figure) 크기를 설정하고 서브플롯 사이와 주변의 여백을 자동으로 조정합니다.
- 엣지, 이름 또는 그래프 속성과 함께 그래프를 초기화합니다.
- 여러 개의 노드를 추가합니다.
add_edges_from()메서드를 사용해 모든 엣지를 한 번에 추가합니다.- Fruchterman-Reingold 힘 기반(force-directed) 알고리즘으로 노드의 좌표를 계산합니다.
- Matplotlib를 사용해 그래프 G를 그립니다.
label_pos값을 지정하여 엣지 레이블을 원하는 위치에 그립니다.show()메서드로 결과 그림을 화면에 표시합니다.
예제 코드
import matplotlib.pylab as plt
import networkx as nx
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
G = nx.DiGraph()
G.add_nodes_from([1, 2, 3, 4])
G.add_edges_from([(1, 2), (2, 3), (3, 4), (4, 1), (1, 3)])
pos = nx.spring_layout(G)
for u, v, d in G.edges(data=True):
d['weight'] = 3
edges, weights = zip(*nx.get_edge_attributes(G, 'weight').items())
nx.draw(G, pos, node_color='b', edge_color=weights, width=2, with_labels=True)
nx.draw_networkx_edge_labels(
G, pos,
{(1, 2): "x", (2, 3): "y", (3, 4): "w",
(4, 1): "z", (1, 3): "v"}, label_pos=0.75
)
plt.show()코드 설명
위 코드에서 핵심 부분은 마지막의 nx.draw_networkx_edge_labels() 호출입니다. 각 엣지에 대응하는 레이블 딕셔너리를 전달하고, label_pos=0.75로 설정함으로써 레이블이 엣지의 중앙이 아니라 시작 노드에서 75% 떨어진 지점, 즉 도착 노드 쪽에 가깝게 배치됩니다. 이 값을 0.5보다 작게 하면 반대로 시작 노드 쪽으로 레이블을 옮길 수 있습니다.
출력 결과
코드를 실행하면 파란색 노드로 구성된 방향 그래프가 그려지고, 각 엣지 위에 'x', 'y', 'w', 'z', 'v' 레이블이 엣지 중심이 아닌 75% 지점에 배치되어 있는 것을 확인할 수 있습니다.