我使用 NetworkX 创建了一个二部图,并想分别为这两组着色。我使用color()
networkX 二分模块中的函数。但是,color dict 中的节点顺序与 B.nodes 中的节点顺序不同,例如:
B.nodes = ['a', 1, 2, 3, 4, 'c', 'b']
二分。颜色(B)= {'a': 1, 1: 0, 2: 0, 'b': 1, 4: 0, 'c': 1, 3: 0}
这会导致图表颜色不正确,如下所示:
代码如下:
B = nx.Graph()
B.add_nodes_from([1,2,3,4], bipartite=0) # Add the node attribute "bipartite"
B.add_nodes_from(['a','b','c'], bipartite=1)
B.add_edges_from([(1,'a'), (1,'b'), (2,'b'), (2,'c'), (3,'c'), (4,'a')])
bottom_nodes, top_nodes = bipartite.sets(B)
color = bipartite.color(B)
color_list = []
for c in color.values():
if c == 0:
color_list.append('b')
else:
color_list.append('r')
# Draw bipartite graph
pos = dict()
color = []
pos.update( (n, (1, i)) for i, n in enumerate(bottom_nodes) ) # put nodes from X at x=1
pos.update( (n, (2, i)) for i, n in enumerate(top_nodes) ) # put nodes from Y at x=2
nx.draw(B, pos=pos, with_labels=True, node_color = color_list)
plt.show()
有什么我想念的吗?
沧海一幻觉
相关分类