From e9c9d08107967119539b3fa7202b3f670868d092 Mon Sep 17 00:00:00 2001 From: lisazeyen <35347358+lisazeyen@users.noreply.github.com> Date: Thu, 11 Apr 2024 09:52:17 +0200 Subject: [PATCH] rename _intervals to _years, shorten definition --- scripts/add_existing_baseyear.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/scripts/add_existing_baseyear.py b/scripts/add_existing_baseyear.py index 75375565..d70afb48 100644 --- a/scripts/add_existing_baseyear.py +++ b/scripts/add_existing_baseyear.py @@ -461,14 +461,12 @@ def add_heating_capacities_installed_before_baseyear( and int(grouping_year) < int(baseyear) ] ) + + # get number of years of each interval + _years = (valid_grouping_years.diff().shift(-1) + .fillna(baseyear-valid_grouping_years.iloc[-1])) # Installation is assumed to be linear for the past - _intervals = ( - pd.concat( - [valid_grouping_years[1:], pd.Series(baseyear)], ignore_index=True - ) - - valid_grouping_years - ) - ratios = _intervals / _intervals.sum() + ratios = _years / _years.sum() for ratio, grouping_year in zip(ratios, valid_grouping_years):