1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
|
"""
=====================
Pygraphviz Attributes
=====================
An example showing how to use the interface to the pygraphviz
AGraph class to convert to and from graphviz.
Also see the pygraphviz documentation and examples at
http://pygraphviz.github.io/
"""
import networkx as nx
# networkx graph
G = nx.Graph()
# ad edges with red color
G.add_edge(1, 2, color="red")
G.add_edge(2, 3, color="red")
# add nodes 3 and 4
G.add_node(3)
G.add_node(4)
# convert to a graphviz agraph
A = nx.nx_agraph.to_agraph(G)
# write to dot file
A.write("k5_attributes.dot")
# convert back to networkx Graph with attributes on edges and
# default attributes as dictionary data
X = nx.nx_agraph.from_agraph(A)
print("edges")
print(list(X.edges(data=True)))
print("default graph attributes")
print(X.graph)
print("node node attributes")
print(X.nodes.data(True))
|