添加具有颜色属性的边/节点

时间:2010-01-29 05:52:19

标签: python networkx

我使用的是networkx Python包。文档说我们可以H.add_edge(1,2,color='blue'),但输出显示默认(黑色)颜色的边。当我H.add_node(12,color='green')时,我得到一个具有相同默认红色的新节点。

1 个答案:

答案 0 :(得分:4)

Peter,根据文档,要更改绘制节点/边缘的颜色,您必须为绘图函数提供node_color参数。即来自this example,绘制这样的图形(注意不同颜色的节点):

alt text http://networkx.lanl.gov/_images/house_with_colors.png

代码是:

#!/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

请注意第二次调用node_color的{​​{1}}参数。这有帮助吗?