python networkx 边权重_python – 基于权重着色networkx边缘

#!/usr/bin/env python

"""

Draw a graph with matplotlib.

You must have matplotlib for this to work.

"""

try:

import matplotlib.pyplot as plt

import matplotlib.colors as colors

import matplotlib.cm as cmx

import numpy as np

except:

raise

import networkx as nx

G=nx.path_graph(8)

#Number of edges is 7

values = range(7)

# These values could be seen as dummy edge weights

jet = cm = plt.get_cmap('jet')

cNorm = colors.Normalize(vmin=0, vmax=values[-1])

scalarMap = cmx.ScalarMappable(norm=cNorm, cmap=jet)

colorList = []

for i in range(7):

colorVal = scalarMap.to_rgba(values[i])

colorList.append(colorVal)

nx.draw(G,edge_color=colorList)

plt.savefig("simple_path.png") # save as png

plt.show() # display

刚刚修改了networkx中的一个示例代码,该代码绘制了一个简单的图形.