AtHeartEngineer/SIMoN
1
0
Fork 0
mirror of https://github.com/JHUAPL/SIMoN.git synced 2026-01-09 14:57:56 -05:00
Code Issues Packages Projects Releases Wiki Activity

some cleanup

Browse Source 
Former-commit-id: 6b42ac9f710c04b42bd9c2a9e661e08037f4df8c
This commit is contained in:
Michael T. Kelbaugh
2020-02-28 14:31:20 -05:00
parent 5c89b3dc05
commit 881b44bc22
1 changed files with 8 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
graphs/build.py
View File

@@ -13,9 +13,6 @@ from networkx.readwrite import json_graph
import geojson
import itertools
import csv
import sys
import time
from collections import defaultdict
import uuid
@@ -45,36 +42,31 @@ states = read_file(f"{shapefile_dir}/shapefiles/simple1000_clipped_state.shp").t
counties = read_file(f"{shapefile_dir}/shapefiles/simple1000_clipped_county.shp").to_crs(epsg=projection)
nercs = read_file(f"{shapefile_dir}/shapefiles/simple1000_clipped_nerc.shp").to_crs(epsg=projection)
huc8s = read_file(f"{shapefile_dir}/shapefiles/simple1000_clipped_huc8.shp").to_crs(epsg=projection)
latlon = read_file(f"{shapefile_dir}/shapefiles/simple1000_clipped_latlon.shp").to_crs(epsg=projection)
latlons = read_file(f"{shapefile_dir}/shapefiles/simple1000_clipped_latlon.shp").to_crs(epsg=projection)
# nerc regions
nercs['country'] = nercs.apply(lambda x: 1, axis=1)
nercs['area'] = nercs.apply(lambda row: row['geometry'].area, axis=1)
nercs['parent_ID'] = 'usa48'
# nerc shapefile has the smallest scope, so it is used to define the scope of the USA
country = nercs.dissolve(by='country',aggfunc={'area': 'sum'})
country['NAME'] = "usa48"
# states
states['parent_ID'] = 'usa48'
# counties
counties.rename(columns={'STATEFP': 'parent_ID'}, inplace=True)
counties['polygons'] = counties.geometry.copy()
counties['county_polygons'] = counties.geometry.copy()
counties.geometry = counties.geometry.representative_point()
counties = sjoin(counties, states, how="inner", op='within')
counties.geometry = counties.polygons
counties.geometry = counties['county_polygons']
counties.drop('index_right', axis=1, inplace=True)
counties.rename(columns={'ID_left': 'ID'}, inplace=True)
counties.rename(columns={'NAME_left': 'NAME'}, inplace=True)
counties.rename(columns={'ID_right': 'parent_ID'}, inplace=True)
# huc8
huc8s['parent_ID'] = 'usa48'
# latitude-longitude grid squares
latlon["parent_ID"] = "usa48"
latlons["parent_ID"] = "usa48"
huc8s['parent_ID'] = 'usa48'
nercs['parent_ID'] = 'usa48'
states['parent_ID'] = 'usa48'
# each shapefile should have 4 attributes: ID, NAME, parent_ID, geometry
shapefiles = {}
@@ -82,7 +74,7 @@ shapefiles["state"] = states
shapefiles["county"] = counties
shapefiles["nerc"] = nercs
shapefiles["huc8"] = huc8s
shapefiles["latlon"] = latlon
shapefiles["latlon"] = latlons
for granularity, shapefile in shapefiles.items():
for column in ['ID', 'NAME', 'geometry', 'parent_ID']: