blob: 5f848a433e4724c3437af683913eb52d75689598 (
plain)
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
39
40
41
42
43
44
45
46
47
48
|
#!/usr/bin/env python
"""
=============
Write Dotfile
=============
Write a dot file from a networkx graph for further processing with graphviz.
You need to have either pygraphviz or pydot for this example.
See https://networkx.github.io/documentation/latest/reference/drawing.html
for more info.
"""
# Author: Aric Hagberg (hagberg@lanl.gov)
# Copyright (C) 2004-2018 by
# Aric Hagberg <hagberg@lanl.gov>
# Dan Schult <dschult@colgate.edu>
# Pieter Swart <swart@lanl.gov>
# All rights reserved.
# BSD license.
import networkx as nx
# and the following code block is not needed
# but we want to see which module is used and
# if and why it fails
try:
import pygraphviz
from networkx.drawing.nx_agraph import write_dot
print("using package pygraphviz")
except ImportError:
try:
import pydot
from networkx.drawing.nx_pydot import write_dot
print("using package pydot")
except ImportError:
print()
print("Both pygraphviz and pydot were not found ")
print("see https://networkx.github.io/documentation/latest/reference/drawing.html")
print()
raise
G = nx.grid_2d_graph(5, 5) # 5x5 grid
write_dot(G, "grid.dot")
print("Now run: neato -Tps grid.dot >grid.ps")
|