script: only aggregate total power values

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)