python如何改颜色_python和networkx：如何更改节点的颜色

我正在使用python和“networkx”来随机分配边缘点。点头在2个类别中，并且每个类别具有相同数量的节点。代码正在工作，但我有一个问题：如何更改节点的颜色？现在，这两个类别的节点都具有sam颜色。python和networkx：如何更改节点的颜色

import numpy as np

import networkx as nx

import matplotlib.pyplot as plt

N = int(raw_input("Enter the number of each race to be used: "))

gamma = float(raw_input("Enter the probability of same-race ties: "))

delta = float(raw_input("Enter the probability of cross-race ties: "))

connectr11 = np.zeros((N,N),dtype=np.int32)

for i in range(0,N):

dval = np.diag((np.random.random_sample(size=(N-i))

connectr11 += dval

if (i>0):

connectr11 += dval.T

##print connectr11

connectr22 = np.zeros((N,N),dtype=np.int32)

for i in range(0,N):

dval = np.diag((np.random.random_sample(size=(N-i))

connectr22 += dval

if (i>0):

connectr22 += dval.T

##print connectr22

connectr12 = np.zeros((N,N),dtype=np.int32)

for i in range(0,N):

dval = np.diag((np.random.random_sample(size=(N-i))

connectr12 += dval

if (i>0):

connectr12 += dval.T

##print connectr12

connectr21 = np.zeros((N,N),dtype=np.int32)

for i in range(0,N):

dval = np.diag((np.random.random_sample(size=(N-i))

connectr12 += dval

if (i>0):

connectr12 += dval.T

##print connectr21

concr1x=np.concatenate((connectr11, connectr12), axis=0)

concrx2=np.concatenate((connectr21, connectr22), axis=0)

concr12=np.concatenate((concr1x, concrx2), axis=1)

##print concr12

print concr12

A=np.matrix(concr12)

G=nx.from_numpy_matrix(A)

for i in range(0,N):

if i < N:

G.node[i] = 1

node_color='b'

else:

G.node[i] = 0

node_color='r'

nx.draw_circular(G, node_color=node_color)

plt.show()

+1

很多在这里与代码图纸的例子：http://networkx.github.io/documentation/latest/gallery.html –