Homec4science
Diffusion solar_potential (master) Browse

solarPOT/Potential_hourly/tilt_aspect_aggregates/commune_batches_gt.py
67c48c8e30edmaster

Last Change
Raw File

commune_batches_gt.py
View Options

import numpy as np
import pandas as pd
import xarray as xr
import sys
import os
import time
# # Define file paths
GT_PATH = sys.argv[1]
AREA_FP = sys.argv[2]
OUTPATH = sys.argv[3]
## CONSTANTS
FLAT_TILT = 10
MAX_ASPECT = 90
# Print file paths
print('Available area file path: %s' %AREA_FP)
print('Gt input path: %s' %GT_PATH)
print('Output path: %s\n\n' %OUTPATH)
# # Load commune map
tt = time.time()
# # Load roof data
roof_info = pd.read_csv(AREA_FP, usecols = ['DF_UID', 'available_area', 'BFS_NUMMER'])
print('Loaded roof info in %.2fs' %(time.time()-tt))
roof_info['suitable'] = (roof_info.available_area >= 8).astype(int)
# # Load prediction
tilted_irrad = xr.open_mfdataset( GT_PATH )
# # Group suitable potential by communes
tt = time.time()
time_xr = tilted_irrad.time
roof_sel = roof_info.loc[ roof_info.DF_UID.isin(tilted_irrad.DF_UID.values),
['DF_UID', 'suitable', 'available_area', 'BFS_NUMMER'] ]
roof_xr = roof_sel.set_index('DF_UID').to_xarray()
hourly_suitable = xr.merge([tilted_irrad.tilted_irradiance, roof_xr.available_area]) * roof_xr.suitable
hourly_suitable.coords['BFS_NUMMER'] = roof_xr.BFS_NUMMER.astype('int')
print(hourly_suitable)
gt_communes_hourly = ( hourly_suitable.tilted_irradiance.where( hourly_suitable.tilted_irradiance > 0 )
.groupby('BFS_NUMMER').mean().astype('float32') )
gt_weighted_sum = ((hourly_suitable.tilted_irradiance * hourly_suitable.available_area )
.groupby('BFS_NUMMER').sum().astype('float32').rename('gt_weighted_by_APV'))
weights_sum = (hourly_suitable.available_area * roof_xr.suitable ).groupby('BFS_NUMMER').sum().rename('APV_weight_sum')
communes_output = xr.merge([gt_communes_hourly, gt_weighted_sum, weights_sum, time_xr]) # gt_communes_hourly
print('\nCreated hourly commune potential in %.2fs\n' %(time.time()-tt))
print(communes_output)
# ### Save intermediate files
tt = time.time()
communes_output.to_netcdf( OUTPATH )
print('\nComputed hourly commune potential and saved %s in %.2fs' %(OUTPATH, time.time()-tt))
c4science · Help