2015-03-29 3 views
0

У меня есть следующий график. Я хочу установить другой цвет узла для мстителей и хранителей галактики, но следующее не работает. Есть ли у вас какие-либо предложения?Как изменить цвет набора узлов?

g = nx.DiGraph() 
g.add_nodes_from(guardians,node_color='b') 
g.add_edges_from(guardians_links) 

g.add_nodes_from(avengers,node_color='y') 
g.add_edges_from(avengers_links) 

plt.figure(figsize=(12,10)) 
nx.draw_random(g) 
plt.show() # display 

ответ

2

заменить nx.draw_random(g) с

pos = nx.random_layout(g) 
nx.draw(g,pos=pos, nodelist = guardians, node_color='b') 
nx.draw(g,pos=pos, nodelist = avengers, node_color='y') 

И вы на самом деле не нужно назначить узел цвета при добавлении узлов.

Вот инструкции по эксплуатации nx.draw. Дополнительные ключевые слова описаны в документации для nx.draw_networkx.

+0

Я получаю это ТипError: draw() получил несколько значений для ключевого слова аргумент 'pos' – dimitrisdan

+0

oops - Я имел в виду, что вызов должен быть в 'nx.draw (...)', а не 'nx.draw_random (. ..) '. См. Отредактированный код. Должно сработать. – Joel

+0

спасибо! это сработало! – dimitrisdan

Смежные вопросы