Skip to content

Instantly share code, notes, and snippets.

@matthiaseisen
Last active May 12, 2018 10:50
Show Gist options
  • Star 19 You must be signed in to star a gist
  • Fork 4 You must be signed in to fork a gist
  • Save matthiaseisen/3278cedcd53afe62c3f3 to your computer and use it in GitHub Desktop.
Save matthiaseisen/3278cedcd53afe62c3f3 to your computer and use it in GitHub Desktop.
import graphviz as gv
g1 = gv.Graph(format='svg')
g1.node('A')
g1.node('B')
g1.edge('A', 'B')
print(g1.source)
filename = g1.render(filename='img/g1')
print(filename)
g2 = gv.Digraph(format='svg')
g2.node('A')
g2.node('B')
g2.edge('A', 'B')
g2.render('img/g2')
import functools
graph = functools.partial(gv.Graph, format='svg')
digraph = functools.partial(gv.Digraph, format='svg')
g3 = graph()
nodes = ['A', 'B', ('C', {})]
edges = [
('A', 'B'),
('B', 'C'),
(('A', 'C'), {}),
]
def add_nodes(graph, nodes):
for n in nodes:
if isinstance(n, tuple):
graph.node(n[0], **n[1])
else:
graph.node(n)
return graph
def add_edges(graph, edges):
for e in edges:
if isinstance(e[0], tuple):
graph.edge(*e[0], **e[1])
else:
graph.edge(*e)
return graph
add_edges(
add_nodes(digraph(), ['A', 'B', 'C']),
[('A', 'B'), ('A', 'C'), ('B', 'C')]
).render('img/g4')
add_edges(
add_nodes(digraph(), [
('A', {'label': 'Node A'}),
('B', {'label': 'Node B'}),
'C'
]),
[
(('A', 'B'), {'label': 'Edge 1'}),
(('A', 'C'), {'label': 'Edge 2'}),
('B', 'C')
]
).render('img/g5')
g6 = add_edges(
add_nodes(digraph(), [
('A', {'label': 'Node A'}),
('B', {'label': 'Node B'}),
'C'
]),
[
(('A', 'B'), {'label': 'Edge 1'}),
(('A', 'C'), {'label': 'Edge 2'}),
('B', 'C')
]
)
styles = {
'graph': {
'label': 'A Fancy Graph',
'fontsize': '16',
'fontcolor': 'white',
'bgcolor': '#333333',
'rankdir': 'BT',
},
'nodes': {
'fontname': 'Helvetica',
'shape': 'hexagon',
'fontcolor': 'white',
'color': 'white',
'style': 'filled',
'fillcolor': '#006699',
},
'edges': {
'style': 'dashed',
'color': 'white',
'arrowhead': 'open',
'fontname': 'Courier',
'fontsize': '12',
'fontcolor': 'white',
}
}
def apply_styles(graph, styles):
graph.graph_attr.update(
('graph' in styles and styles['graph']) or {}
)
graph.node_attr.update(
('nodes' in styles and styles['nodes']) or {}
)
graph.edge_attr.update(
('edges' in styles and styles['edges']) or {}
)
return graph
g6 = apply_styles(g6, styles)
g6.render('img/g6')
g7 = add_edges(
add_nodes(digraph(), [
('A', {'label': 'Node A'}),
('B', {'label': 'Node B'}),
'C'
]),
[
(('A', 'B'), {'label': 'Edge 1'}),
(('A', 'C'), {'label': 'Edge 2'}),
('B', 'C')
]
)
g8 = apply_styles(
add_edges(
add_nodes(digraph(), [
('D', {'label': 'Node D'}),
('E', {'label': 'Node E'}),
'F'
]),
[
(('D', 'E'), {'label': 'Edge 3'}),
(('D', 'F'), {'label': 'Edge 4'}),
('E', 'F')
]
),
{
'nodes': {
'shape': 'square',
'style': 'filled',
'fillcolor': '#cccccc',
}
}
)
g7.subgraph(g8)
g7.edge('B', 'E', color='red', weight='2')
g7.render('img/g7')
@jim-snyder-grant
Copy link

Very helpful! But for a newbie like me, calling the sample file graphviz.py was a bit confusing at first. After stumbling around a bit, I saw I was attempting to import the sample file itself, instead of the real graphviz. Might want to rename the sample file..

@RexYuan
Copy link

RexYuan commented Nov 20, 2017

Thanks for the tutorial. Did you figure out a way to stylize final states (like double circle)?

edit: found the doc for it: use attr method and shape kwarg.
edit2: now I wonder how to stylize initial states, namely to get an edge without predecessor to point at it?

@JamesWharton
Copy link

Hey this is really cool. I am trying to figure out how to a couple of things.
How to deal with rendering multiple subgraphs ?
How to create a subgraph border.

I am not sure also what diffence is between a subgraph and a cluster?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment