python加颜色_python – 向网络图添加图例以解释节点的颜色

我有一个networkx图的图,其中边缘颜色取决于使用以下代码分配给相应边的权重(使用a_netw nx.Graph)：

a_netw_edges = a_netw.edges()

a_netw_weights = [a_netw[source][dest]['weight'] for source, dest in a_netw_edges]

a_netw_colors = [plt.cm.Blues(weight*15) for weight in a_netw_weights]

nx.draw_networkx(a_netw, edges=a_netw_edges, width=1, edge_color=a_netw_colors)

在这张图中,我想添加一个图例,使权重和颜色之间的连接显式;就像在使用pcolor的热图中一样.

虽然我对如何开始有一个粗略的想法：

fig, axes = plt.subplots(nrows=2)

nx.draw_networkx(a_netw, edges=a_netw_edges, width=1, edge_color=a_netw_colors, ax=axes[0])

axes[0].get_xaxis().set_visible(False)

axes[0].get_yaxis().set_visible(False)

gradient = np.linspace(0, 1, 256)

gradient = np.vstack((gradient, gradient))

axes[1].imshow(gradient, aspect=3, cmap=plt.cm.Blues)

axes[1].get_yaxis().set_visible(False)

plt.tight_layout()