Tô màu các cạnh theo trọng lượng trong networkx (Matplotlib)

Để tô màu các cạnh theo trọng lượng trong networkx , chúng tôi có thể thực hiện các bước sau -

• Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
• Khởi tạo một biểu đồ với các thuộc tính cạnh, tên hoặc biểu đồ.
• Thêm các nút vào biểu đồ hiện tại.
• Thêm các cạnh vào các nút của biểu đồ hiện tại.
• Lặp lại các cạnh của biểu đồ đã cho và đặt một số trọng số cho chúng.
• Vẽ biểu đồ hiện tại với trọng số cho màu sắc cạnh.
• Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import random as rd
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)])
for u, v, d in G.edges(data=True):
   d['weight'] = rd.random()
edges, weights = zip(*nx.get_edge_attributes(G, 'weight').items())
nx.draw(G, node_color='b', edge_color=weights, width=2, with_labels=True)

plt.show()

Đầu ra