添加带颜色属性的边/节点

Question

我正在使用Python的networkx库。文档上说我们可以用H.add_edge(1,2,color='blue')来添加一条边，并指定颜色为蓝色，但输出却显示边的颜色是默认的（黑色）。当我用H.add_node(12,color='green')添加一个新节点时，结果却是这个新节点的颜色还是默认的红色。

Answer 1

彼得，根据文档，要改变节点和边的颜色，你需要在绘图函数中提供node_color这个参数。也就是说，从这个例子来看，要绘制一个像这样的图（注意节点的颜色不同）：

代码是：

#!/usr/bin/env python
"""
Draw a graph with matplotlib.
You must have matplotlib for this to work.
"""
__author__ = """Aric Hagberg (hagberg@lanl.gov)"""
try:
    import matplotlib.pyplot as plt
except:
    raise

import networkx as nx

G=nx.house_graph()
# explicitly set positions
pos={0:(0,0),
     1:(1,0),
     2:(0,1),
     3:(1,1),
     4:(0.5,2.0)}

nx.draw_networkx_nodes(G,pos,node_size=2000,nodelist=[4])
nx.draw_networkx_nodes(G,pos,node_size=3000,nodelist=[0,1,2,3],node_color='b')
nx.draw_networkx_edges(G,pos,alpha=0.5,width=6)
plt.axis('off')
plt.savefig("house_with_colors.png") # save as png
plt.show() # display

注意在第二次调用draw_networkx_nodes时使用了node_color这个参数。这样解释清楚了吗？