bjoern/pypsa-eur
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

add_electricity: make drop_leap_year own function

Browse Source
This commit is contained in:
Fabian Neumann 2022-06-16 09:00:33 +02:00
parent e9bab1622c
commit 173ab7b318
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/add_electricity.py
View File

@ -547,6 +547,15 @@ def add_nice_carrier_names(n, config):
logger.warning(f'tech_colors for carriers {missing_i} not defined in config.') logger.warning(f'tech_colors for carriers {missing_i} not defined in config.')
n.carriers['color'] = colors n.carriers['color'] = colors
def drop_leap_day(n):
if not n.snapshots.is_leap_year.any(): return
leap_days = (n.snapshots.day == 29) & (n.snapshots.month == 2)
n.set_snapshots(n.snapshots[~leap_days])
n.snapshot_weightings[:] = 8760/len(n.snapshots)
logger.info("Dropped February 29 from leap year.")
if __name__ == "__main__": if __name__ == "__main__":
if 'snakemake' not in globals(): if 'snakemake' not in globals():
from _helpers import mock_snakemake from _helpers import mock_snakemake
@ -600,7 +609,6 @@ if __name__ == "__main__":
add_nice_carrier_names(n, snakemake.config) add_nice_carrier_names(n, snakemake.config)
if snakemake.config['enable'].get('drop_leap_days', True): if snakemake.config['enable'].get('drop_leap_days', True):
leap_days = (n.snapshots.day == 29) & (n.snapshots.month == 2) drop_leap_day(n)
n.set_snapshots(n.snapshots[~leap_days])
n.export_to_netcdf(snakemake.output[0]) n.export_to_netcdf(snakemake.output[0])