Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách tùy chỉnh hiển thị nhãn cạnh (edge labels) dùng networkx trong Matplotlib?

Để đặt offset nhãn cạnh networkx:

  • Thiết lập kích thước hình và điều chỉnh khoảng đệm subplot.
  • Khởi tạo đồ thị với cạnh, tên, thuộc tính.
  • Thêm nhiều node.
  • Định vị node bằng thuật toán Fruchterman-Reingold force-directed.
  • Vẽ đồ thị G với Matplotlib.
  • Vẽ nhãn cạnh.
  • Hiển thị hình bằng show().

Ví dụ

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()

Kết quả

Cách tùy chỉnh hiển thị nhãn cạnh (edge labels) dùng networkx trong Matplotlib?