Двухцветный цвет NetworkX смешивает порядок - PullRequest
Новая
       35

Двухцветный цвет NetworkX смешивает порядок

0 голосов
/ 14 февраля 2019

Я создал двудольный график с использованием NetworkX и хотел бы раскрасить два набора по отдельности.Я использую функцию color() из двустороннего модуля networkX.Тем не менее, порядок узлов отличается в цветовом диктанте, чем в B.nodes, например:

B.nodes = ['a', 1, 2, 3, 4, 'c', 'b']

bipartite.color (B) = {'a': 1, 1: 0, 2: 0, 'b': 1, 4: 0, 'c': 1, 3: 0}

Это приводит к тому, что график неправильно окрашен, как показано ниже:

incorrectly colour graph

Код выглядит следующим образом: 

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()

Что-то мне не хватает?

Спасибо.

1 Ответ

0 голосов
/ 14 февраля 2019

Ваш список color_list и список узлов (B.nodes) находятся в другом порядке при рисовании графика. 

color_list
['r', 'b', 'b', 'r', 'b', 'r', 'r']

B.nodes
NodeView((1, 2, 3, 4, 'a', 'b', 'c'))

Я создал список color_list с использованием порядка B.nodes, используя словарь и отображая двудольные наборы изсписок узлов в B. 

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_dict = {0:'b',1:'r'}

color_list = [color_dict[i[1]] for i in B.nodes.data('bipartite')]

# 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()

Вывод:

enter image description here

...