merge_horizon_2m.py
b9297dda7962aw_solar

merge_horizon_2m.py
View Options

			import numpy as np
			import pandas as pd
			import os
			import sys
			import time

			############ USER INPUT #########

			# location of csv files with region-based shading stats (without .csv ending)
			shading_base = sys.argv[1]

			# total number of regions
			n_regions = int(sys.argv[2])

			SEPARATOR = '\|'
			ZONE_NAME = 'zone'

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

			shading_all = []

			tt = time.time()
			for region in range(n_regions):

			# load shading file
			filename = ('%s_region%d.csv' %(shading_base, region))
			new_shading_file = pd.read_csv(filename, sep = SEPARATOR )
			print('Loaded file %s' %(filename))

			shading_all.append(new_shading_file)

			# get time:
			print('Read file in %.2f seconds' %(time.time()-tt))
			tt = time.time()

			shading_all = pd.concat(shading_all).sort_values([ZONE_NAME]).reset_index().drop('index', axis=1)

			print('Loaded all files, concatenation in %.2f seconds\n' %(time.time()-tt))

			# make sure every roof only exists once - for simplicity, take average on roofs that exist multiplte times (only ver small percentage anyways)
			shading_all_noDuplicates = shading_all.groupby(ZONE_NAME).mean()

			# save as csv
			filename = '%s.csv' %(shading_base)
			shading_all_noDuplicates.to_csv(filename)
			print('Saved csv to %s' %filename)