diff --git a/scripts-py/expe_energumen/m100_agg_month_power_values.py b/scripts-py/expe_energumen/m100_agg_month_power_values.py index 27d43e622216cb9a4eb18ad45833709ba02135c8..136179870e4849f8b43772dbb5b2275aacf48d32 100644 --- a/scripts-py/expe_energumen/m100_agg_month_power_values.py +++ b/scripts-py/expe_energumen/m100_agg_month_power_values.py @@ -28,30 +28,12 @@ def several_months(): parser.add_argument("month", nargs='+', help="the month to aggregate. example value: '22-07'") args = parser.parse_args() - full_p0_df = None - full_p1_df = None full_total_df = None for month in args.month: month_prefix = f'{args.input_prefix}{month}' - p0_file = f'{month_prefix}_power_p0.parquet' - p1_file = f'{month_prefix}_power_p1.parquet' total_file = f'{month_prefix}_power_total.parquet' - p0_df = read_aggregate_month(p0_file) - if full_p0_df is None: - full_p0_df = p0_df - else: - full_p0_df = pd.concat([full_p0_df, p0_df]) - del p0_df - - p1_df = read_aggregate_month(p1_file) - if full_p1_df is None: - full_p1_df = p1_df - else: - full_p1_df = pd.concat([full_p1_df, p1_df]) - del p1_df - total_df = read_aggregate_month(total_file) if full_total_df is None: full_total_df = total_df @@ -59,10 +41,5 @@ def several_months(): full_total_df = pd.concat([full_total_df, total_df]) del total_df - p0_agg = full_p0_df.groupby(['power', 'node']).sum().reset_index() - p1_agg = full_p1_df.groupby(['power', 'node']).sum().reset_index() total_agg = full_total_df.groupby(['power', 'node']).sum().reset_index() - - p0_agg.to_csv(f'{args.output_prefix}power_p0.csv', index=False) - p1_agg.to_csv(f'{args.output_prefix}power_p1.csv', index=False) total_agg.to_csv(f'{args.output_prefix}power_total.csv', index=False)