Define function `read_eddy_graph`

Analysis/report_graph.py

@@ -4,20 +4,24 @@ import networkx as nx
 import csv
 import shapefile
 
-G = nx.DiGraph()
+def read_eddy_graph():
+    G = nx.DiGraph()
 
-with open("edgelist.csv") as f:
-    reader = csv.reader(f, delimiter = " ", skipinitialspace = True)
+    with open("edgelist.csv") as f:
+        reader = csv.reader(f, delimiter = " ", skipinitialspace = True)
 
-    # Skip title lines:
-    next(reader) 
-    next(reader)
-    
-    for row in reader:
-        k1, i1, k2, i2, weight = int(row[0]), int(row[1]), int(row[2]), \
-                                 int(row[3]), float(row[4])
-        G.add_edge((k1, i1), (k2, i2), weight = weight)
+        # Skip title lines:
+        next(reader) 
+        next(reader)
 
+        for row in reader:
+            k1, i1, k2, i2, weight = int(row[0]), int(row[1]), int(row[2]), \
+                                     int(row[3]), float(row[4])
+            G.add_edge((k1, i1), (k2, i2), weight = weight)
+
+    return G
+
+G = read_eddy_graph()
 print("Number of nodes:", len(G))
 print("Number of edges:", G.number_of_edges())
 n1 = 0