Homec4science
Diffusion solar_potential (master) Browse

solarPOT/Panel_fitting/postprocess_panelled_area.py
591b5c66d542master

Last Change
Raw File

postprocess_panelled_area.py
View Options

import pandas as pd
import geopandas as gpd
import os
import sys
# Define file paths
horiz_fp = sys.argv[1]
vert_fp = sys.argv[2]
best_fp = sys.argv[3]
if len(sys.argv) > 4:
SAVE_SHAPE = True
shape_name = sys.argv[4]
else:
SAVE_SHAPE = False
# Read inputs
stats_horizontal = pd.read_csv(horiz_fp)
stats_vertical = pd.read_csv(vert_fp)
## Merge data from horizontal and vertical panel fitting
stats_both = stats_horizontal.loc[:,['DF_UID', 'FLAECHE', 'panel_tilt', 'NEIGUNG']]
stats_both['count_vert'] = stats_vertical.panel_count
stats_both['count_horiz'] = stats_horizontal.panel_count
stats_both.loc[stats_both.count_vert > stats_both.count_horiz, 'best_align'] = 'vertical'
stats_both.loc[stats_both.count_vert < stats_both.count_horiz, 'best_align'] = 'horizontal'
stats_both.loc[stats_both.count_vert == stats_both.count_horiz, 'best_align'] = 'equal'
## Create a dataframe with optimal alignment
stats_best = stats_both.loc[:, ['DF_UID', 'panel_tilt', 'best_align']]
stats_best['panel_count'] = stats_horizontal.panel_count
stats_best['panelled_area'] = stats_horizontal.panelled_area
stats_best['panelled_area_ratio'] = stats_horizontal.panelled_area_ratio
idx_vertical_align = stats_both.best_align == 'vertical'
stats_best.loc[idx_vertical_align, 'panel_count'] = stats_vertical.loc[idx_vertical_align, 'panel_count'].values
stats_best.loc[idx_vertical_align, 'panelled_area'] = stats_vertical.loc[idx_vertical_align, 'panelled_area'].values
stats_best.loc[idx_vertical_align, 'panelled_area_ratio'] = stats_vertical.loc[idx_vertical_align, 'panelled_area_ratio'].values
# Save output
stats_best.to_csv(best_fp, index = False)
print('Saved %s' %best_fp)
# Return if shapes are not processed
if not SAVE_SHAPE:
exit()
# Extract filepaths for shapes
shape_horiz_path = os.path.split(horiz_fp)[0]
shape_verti_path = os.path.split(vert_fp) [0]
shape_best_path = os.path.split(best_fp) [0]
# Read shapefiles
shapes_horiz = gpd.read_file(os.path.join(shape_horiz_path, shape_name))
shapes_verti = gpd.read_file(os.path.join(shape_verti_path, shape_name))
# Extract best-fitting panels (horizontal if equal)
horiz_UUID = stats_best[stats_best.best_align.isin(['horizontal', 'equal'])].DF_UID
shapes_best_horiz = shapes_horiz[shapes_horiz.DF_UID.isin( horiz_UUID )]
verti_UUID = stats_best[stats_best.best_align== 'vertical'].DF_UID
shapes_best_verti = shapes_verti[shapes_verti.DF_UID.isin( verti_UUID )]
# Append both and save output
shapes_best = gpd.GeoDataFrame( pd.concat([shapes_best_horiz, shapes_best_verti]) )
shapes_best.crs = shapes_horiz.crs
shapes_best.to_file( os.path.join(shape_best_path, shape_name) )
print('Saved %s' %os.path.join(shape_best_path, shape_name))
c4science · Help