python draw函数_networkx draw函数的图例

所以我有下面的函数来画一个我正在处理的问题。它基本上是一个关键节点检测问题或阻断。我有一些值x，并决定攻击节点z。基本上我想用活动和非活动节点以及正在处理/攻击的节点来着色我的图。这是我到目前为止的情况。在

def draw_solution(g，zsolution，xsolution，T，xmin=0，filename='测试.pdf')：# draw solution

pos = {n: ndata['coord'] for n,ndata in g.nodes_iter(data=True)}

ncolour=[]

for n,ndata in g.nodes_iter(data=True):

if ndata['fuel_load'] < xmin:

ncolour.append('gold')

else:

ncolour.append('yellowgreen')

pp = PdfPages(filename)

fig = plt.figure(figsize=(11.7,8.3))

fig.suptitle('full graph and initial fuel load')

plt.axis('off')

nx.draw_networkx(g, pos, font_size=9, node_color=ncolour)

fig.savefig(pp, format='pdf')

fig.clf()

for t in range(T):

g_copy = g.copy()

#node colour

ncolour=[]

for i in g.nodes_iter():

if zsolution[i,t] > 0.99:

ncolour.append('lightcoral')

#g_copy.remove_edges_from(g.edges(i))

elif xsolution[i,t] < xmin:

ncolour.append('gold')

g_copy.remove_edges_from(g.edges(i))

else:

ncolour.append('yellowgreen')

plt.axis('off')

fig.suptitle('t={}, before treatment'.format(t))

nx.draw_networkx(g_copy, pos, font_size=9, node_color=ncolour, label=ncolour)

fig.savefig(pp, format='pdf')

fig.clf()

g_copy = g.copy()

#node colour

ncolour=[]

for i in g.nodes_iter():

if zsolution[i,t] > 0.99:

ncolour.append('lightcoral')

g_copy.remove_edges_from(g.edges(i))

elif xsolution[i,t] < xmin:

ncolour.append('gold')

g_copy.remove_edges_from(g.edges(i))

else:

ncolour.append('yellowgreen')

plt.axis('off')

fig.suptitle('t={}, after treatment'.format(t))

nx.draw_networkx(g_copy, pos, font_size=9, node_color=ncolour, label=ncolour)

fig.savefig(pp, format='pdf')

fig.clf()

pp.close()

fig.clf()

因为某些原因我不能正确地理解传说？在

请帮忙！谢谢！在