diff --git a/Analysis/report_graph.py b/Analysis/report_graph.py index b40971e36a06d3ce92ec5d8962c3bb0eac3b303d..3d5c21b053a1f121790930430e9e24fff654dedd 100755 --- a/Analysis/report_graph.py +++ b/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