在 networkx 中按权重为边着色(Matplotlib)
matplotlibpythondata visualization
要在 networkx 中按权重为边着色，我们可以采取以下步骤 −
- 设置图形大小并调整子图之间和周围的填充。
- 使用边、名称或图形属性初始化图形。
- 向当前图形添加节点。
- 向当前图形的节点添加边。
- 迭代给定图形的边并为它们设置一些权重。
- 使用权重绘制当前图形作为边颜色。
- 要显示图形，请使用 show() 方法。
示例
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()
输出
