I have a graph G with attribute 'state' for nodes and edges. I want to draw the graph, all nodes labelled, and with the state marked outside the corresponding edge/node.
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'
The command draw.networkx has an option for labels, but I do not understand how to provide the attribute as a label to this command. Could someone help me out?
It's not so pretty - but it works like this:
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()
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With