Added functions to compare policies

src/utils/imbalance_price_calculator.py

@@ -1,6 +1,6 @@
-from datetime import datetime
 import plotly.graph_objects as go
 import numpy as np
+import pytz
 import pandas as pd
 
 incremental_bids = "../../data/incremental_bids.csv"
@@ -66,8 +66,6 @@ class ImbalancePriceCalculator:
         dec_bids = row["bid_ladder_dec"].values[0]
         inc_bids = row["bid_ladder_inc"].values[0]
         
-
-
         # Prepare data for plot
         x_inc_interpolated = [vol for i in range(len(inc_bids) - 1) for vol in [inc_bids[i][0], inc_bids[i+1][0]]]
         y_inc_interpolated = [price for cum_vol, price in inc_bids for _ in (0, 1)]
@@ -90,6 +88,14 @@ class ImbalancePriceCalculator:
 
         fig.show()
 
+    def get_imbalance_prices_2023_for_date(self, date, NRV_predictions):
+        imbalance_prices = []
+        for i in range(1, len(NRV_predictions)):
+            datetime = date + pd.Timedelta(hours=i-1)
+            datetime = pytz.utc.localize(datetime)
+            imbalance_prices.append(self.get_imbalance_price_2023(datetime, NRV_predictions[i-1], NRV_predictions[i]))
+        return [x[1] for x in imbalance_prices]
+
     def get_imbalance_price_2023(self, datetime, NRV_PREV, NRV):
         MIP = self.get_imbalance_price(datetime, abs(NRV))
         MDP = self.get_imbalance_price(datetime, -abs(NRV))