continued bug fixes and graph visualizations

gcrsef-gang · Jan 28, 2023 · 9128c9a · 9128c9a
1 parent 37164e6
commit 9128c9a
Show file tree

Hide file tree

Showing 10 changed files with 135 additions and 70 deletions.
diff --git a/2010_california_geodata.png b/2010_california_geodata.png
diff --git a/2010_florida_geodata.png b/2010_florida_geodata.png
diff --git a/2010_illinois_geodata.png b/2010_illinois_geodata.png
diff --git a/2010_kentucky_geodata.png b/2010_kentucky_geodata.png
diff --git a/2010_maine_geodata.png b/2010_maine_geodata.png
diff --git a/2010_maryland_geodata.png b/2010_maryland_geodata.png
diff --git a/before_connected_2010_vermont_geodata.png b/before_connected_2010_vermont_geodata.png
diff --git a/graph_visualization.py b/graph_visualization.py
@@ -110,6 +110,8 @@ def visualize_graph(graph, output_path, coords, colors=None, edge_colors=None, s
     modified_coords = modify_coords(
         [coords(node) for node in graph_nodes], [2000, 2000]
     )
+
+    print("Coordinates modified")
 
     if colors is not None:
         node_colors = [colors(node) for node in graph_nodes]
@@ -140,13 +142,17 @@ def visualize_graph(graph, output_path, coords, colors=None, edge_colors=None, s
             width=1
     )
 
+    print("Edges drawn")
+
     for center, node, color, r in zip(modified_coords, graph_nodes,
                                       node_colors, node_sizes):
         draw.ellipse(
             [(center[0] - r, center[1] - r),
              (center[0] + r, center[1] + r)],
             fill=color
         )
+
+    print("Nodes drawn")
     # y_coords = [42.72685, 42.848564, 43.022452, 43.213817, 43.354144, 43.533341, 43.607655, 43.750837, 43.873348, 44.008176, 44.107845, 44.185469, 44.304375, 44.382911, 44.461759, 44.492504, 44.57663, 44.670759, 44.781578, 44.858507]
     # y_coords = [shapely.geometry.Point((-73, coord)) for coord in y_coords]
     # new_y_coords = modify_coords(y_coords, [2000,2000])

diff --git a/serialize.py b/serialize.py
diff --git a/temp.py b/temp.py
@@ -0,0 +1,11 @@
+import geopandas as gpd
+
+x = gpd.read_file("good_target_geometry.json")
+post = gpd.read_file("post_target_geometry.json")
+x = x.set_index("geoid")
+post = post.set_index("geoid")
+x_set = set(x.index)
+post_set = set(post.index)
+diff = post_set.difference(x_set)
+geoid = post.loc[diff]
+print(geoid)