03_compute_elevations.py

#!/usr/bin/python3

# This script computes the the elevation data of all the Traffic Analysis Zones
# (TAZ)

import os
import glob
import zipfile
import pandas as pd
import geopandas as gpd
from stplanpy import srtm

################################################################################

in_dir  = os.path.expanduser("../pct-inputs/01_raw/03_elevation/")
out_dir = os.path.expanduser("../pct-inputs/02_intermediate/")

os.makedirs(out_dir, exist_ok=True)

################################################################################

# Read TAZ centroids
taz_cent = pd.read_pickle(out_dir + "01_geographies/bayArea_taz_cent.pkl")

# Create DataFrame with elevations
taz_cent = taz_cent.elev(in_dir + "srtm_12_05.zip")

# Write to pickle
taz_cent.to_pickle(out_dir + "01_geographies/bayArea_taz_cent.pkl")

################################################################################