Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Matplotlib에서 networkx를 사용하여 에지 레이블 표시를 어떻게 사용자 정의합니까?

<시간/>

networkx를 설정하려면 가장자리 레이블 오프셋, 우리는 다음 단계를 수행할 수 있습니다 -

  • 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
  • 가장자리, 이름 또는 그래프 속성을 사용하여 그래프를 초기화합니다.
  • 여러 노드를 추가합니다.
  • Fruchterman-Reingold force-directed 알고리즘을 사용하여 노드를 배치합니다.
  • Matplotlib로 그래프 G를 그립니다.
  • 가장자리 레이블을 그립니다.
  • 그림을 표시하려면 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()

출력

Matplotlib에서 networkx를 사용하여 에지 레이블 표시를 어떻게 사용자 정의합니까?