2013-12-04 4 views
12

У меня есть граф G с атрибутом 'state' для узлов и ребер. Я хочу нарисовать график, все отмеченные узлы и состояние, отмеченное вне соответствующего ребра/узла.Networkx: как показать атрибуты узла и края в графическом чертеже

for v in G.nodes():  
    G.node[v]['state']='X' 
G.node[1]['state']='Y' 
G.node[2]['state']='Y' 

for n in G.edges_iter():  
    G.edge[n[0]][n[1]]['state']='X' 
G.edge[2][3]['state']='Y' 

Команда draw.networkx имеет опцию для этикеток, но я не понимаю, как указать атрибут в качестве метки для этой команды. Может ли кто-нибудь помочь мне?

ответ

21

Это не так красиво - но это работает так:

from matplotlib import pyplot as plt 
import networkx as nx 
G = nx.Graph() 
G.add_edge(1,2) 
G.add_edge(2,3) 
for v in G.nodes(): 
    G.node[v]['state']='X' 
G.node[1]['state']='Y' 
G.node[2]['state']='Y' 

for n in G.edges_iter(): 
    G.edge[n[0]][n[1]]['state']='X' 
G.edge[2][3]['state']='Y' 

pos = nx.spring_layout(G) 

nx.draw(G, pos) 
node_labels = nx.get_node_attributes(G,'state') 
nx.draw_networkx_labels(G, pos, labels = node_labels) 
edge_labels = nx.get_edge_attributes(G,'state') 
nx.draw_networkx_edge_labels(G, pos, labels = edge_labels) 
plt.savefig('this.png') 
plt.show() 

enter image description here

+0

Спасибо, но мне нужно небольшое изменение. Ярлыки узлов отличаются от состояний. Мне нужны X и Y поверх узлов и числа, подобные 1 и 2 внутри них. – Bravo

+1

Если вам нужен более сложный (и более красивый) рисунок меток, вы можете использовать Graphviz для рисования - graphivz.org. Вывод вашего графика из networkx (в том числе с атрибутами, если вы хотите) в формат точки с помощью write_dot, а затем обработать это с помощью Graphviz. – Aric

+1

Позиции всех узлов находятся в словаре «pos». Вы можете пометить их вручную, используя, например, x, y = pos [1], а затем plt.text (x, y, 'label') – Aric

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