Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีปรับแต่งการแสดง Edge Labels ใน NetworkX ด้วย Matplotlib

การปรับตำแหน่ง Edge Labels ใน NetworkX Graph

ใช้ label_pos parameter ใน draw_networkx_edge_labels()

ขั้นตอนหลัก

  • ตั้งค่าขนาดรูปและ padding
  • สร้าง DiGraph และเพิ่ม nodes, edges
  • ตั้งค่า weight attributes
  • คำนวณ layout ด้วย spring_layout
  • วาด graph ด้วย nx.draw()
  • วาด edge labels ด้วย nx.draw_networkx_edge_labels() ระบุ label_pos=0.75
  • แสดงผลด้วย 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()

คำอธิบาย

label_pos=0.75 วาง label ที่ 75% ของความยาว edge จาก source ไป target