Final changes to figures

Author: staadecker

papers/Martin_Staadecker_et_al_2022/analysis.py

@@ -32,10 +32,17 @@
 prebuilt = prebuilt[(prebuilt.build_year + prebuilt.gen_max_age) > 2051]
 print("prebuilt alive :", len(prebuilt))
 
+print("prebuild by tech")
+prebuilt_by_tech = prebuilt.groupby(
+    ["gen_energy_source", "gen_tech"]
+).GENERATION_PROJECT.count()
+print(prebuilt_by_tech)
+
 prebuilt_by_tech = (
     prebuilt.groupby(["gen_energy_source", "gen_tech"]).gen_capacity_limit_mw.sum()
     / 1000
 )
+print("prebuilt by tech capacity")
 print(prebuilt_by_tech.sort_values(ascending=False))
 print(prebuilt_by_tech.sum())
 

papers/Martin_Staadecker_et_al_2022/figure-1-baseline.py

@@ -149,11 +149,11 @@ def rolling_avg(df):
         color=colors[columnName],
         label=columnName + " (no curtail.)",
     )
-ax_right.plot(duals, label="Estimated LMP", color="red")
+ax_right.plot(duals, label="Marginal Price", color="red")
 lines += ax.plot(load, color="orange", label="Demand")
 ax.set_title("A. Seasonal Profiles in the Baseline")
 ax.set_ylabel("Dispatch (TWh/day)")
-ax_right.set_ylabel("Estimated Locational Marginal Price ($/MWh)")
+ax_right.set_ylabel("Marginal Price of Electricity ($/MWh)")
 locator = mdates.MonthLocator()
 ax.xaxis.set_major_formatter(mdates.ConciseDateFormatter(locator))
 ax.set_ylim(-0.1, 4.7)

papers/Martin_Staadecker_et_al_2022/figure-s5-map-of-load-zones.py

@@ -0,0 +1,55 @@
+"""
+Creates a map of the WECC with each load zone labelled with it's SWITCH name.
+Useful for figuring out what load zone name matches what physical region.
+The map is available in the REAM Google Drive under Research -> Switch documentation.
+"""
+
+from matplotlib import pyplot as plt
+
+from papers.Martin_Staadecker_et_al_2022.util import (
+    get_scenario,
+    set_style,
+    save_figure,
+)
+from switch_model.tools.graph.main import GraphTools
+
+tools = GraphTools([get_scenario("1342")], set_style=False)
+tools.pre_graphing(multi_scenario=False)
+
+# %% CREATE PLOT FRAME
+set_style()
+plt.close()
+fig = plt.figure()
+fig.set_size_inches(6.850394, 6.850394)
+ax = fig.add_subplot(1, 1, 1, projection=tools.maps.get_projection())
+
+tools.maps.draw_base_map(ax)
+
+centers = {}
+
+for _, lz in tools.maps._center_points.iterrows():
+    center = lz.geometry.centroid
+    ax.scatter(center.x, center.y, color="k", s=5, alpha=0.5)
+    ax.text(center.x, center.y, lz.gen_load_zone, fontsize="small")
+    centers[lz.gen_load_zone] = center
+
+tx = tools.get_dataframe("transmission_lines.csv", from_inputs=True)
+tx = tx[["trans_lz1", "trans_lz2"]]
+for _, line in tx.iterrows():
+    from_center = centers[line["trans_lz1"]]
+    to_center = centers[line["trans_lz2"]]
+    ax.plot(
+        [from_center.x, to_center.x],
+        [from_center.y, to_center.y],
+        color="k",
+        linestyle="--",
+        linewidth=1,
+        alpha=0.3,
+    )
+
+for lz, center in centers.items():
+    ax.text(center.x, center.y, lz, fontsize="small")
+
+plt.tight_layout()
+# %%
+save_figure("figure-s5-map-of-load-zones.png")

papers/Martin_Staadecker_et_al_2022/figure-s5-duration-cdf-cost-scenarios.py renamed to papers/Martin_Staadecker_et_al_2022/figure-x1-duration-cdf-cost-scenarios.py

@@ -4,24 +4,28 @@
 
 from papers.Martin_Staadecker_et_al_2022.util import (
     set_style,
-    get_scenario, save_figure
+    get_scenario,
+    save_figure,
 )
 from switch_model.tools.graph.main import GraphTools
 
 # Prepare graph tools
-tools = GraphTools(scenarios=[
-    get_scenario("C21", 0.5),
-    get_scenario("C18", 1),
-    get_scenario("C22", 2),
-    get_scenario("C23", 5),
-    get_scenario("C26", 7),
-    get_scenario("C17", 10),
-    get_scenario("C24", 15),
-    get_scenario("1342", 22.43),
-    get_scenario("C25", 40),
-    get_scenario("C19", 70),
-    get_scenario("C20", 102)
-], set_style=False)
+tools = GraphTools(
+    scenarios=[
+        get_scenario("C21", 0.5),
+        get_scenario("C18", 1),
+        get_scenario("C22", 2),
+        get_scenario("C23", 5),
+        get_scenario("C26", 7),
+        get_scenario("C17", 10),
+        get_scenario("C24", 15),
+        get_scenario("1342", 22.43),
+        get_scenario("C25", 40),
+        get_scenario("C19", 70),
+        get_scenario("C20", 102),
+    ],
+    set_style=False,
+)
 tools.pre_graphing(multi_scenario=True)
 
 set_style()
@@ -32,7 +36,7 @@
 ax.clear()
 duration = tools.get_dataframe("storage_capacity.csv")
 duration["duration"] = duration["duration"] = (
-        duration["OnlineEnergyCapacityMWh"] / duration["OnlinePowerCapacityMW"]
+    duration["OnlineEnergyCapacityMWh"] / duration["OnlinePowerCapacityMW"]
 )
 
 duration = duration.sort_values("duration")
@@ -44,22 +48,60 @@
     duration_scenario["cuml_power"] = duration_scenario.OnlinePowerCapacityMW.cumsum()
     duration_scenario = duration_scenario.set_index("cuml_power")
     duration_scenario = duration_scenario["duration"]
-    line = ax.plot(duration_scenario.index, duration_scenario, drawstyle="steps", label=scenario_name)
+    line = ax.plot(
+        duration_scenario.index,
+        duration_scenario,
+        drawstyle="steps",
+        label=scenario_name,
+    )
     if float(int(scenario_name)) == scenario_name:
         label = str(int(scenario_name))
     else:
         label = str(scenario_name)
-    labellines.labelLine(line[0], duration_scenario.index.max(), outline_width=2, label=label + "$/KWh", align=False,
-                         fontsize="small")
+    labellines.labelLine(
+        line[0],
+        duration_scenario.index.max(),
+        outline_width=2,
+        label=label + "$/KWh",
+        align=False,
+        fontsize="small",
+    )
 
 ax.set_yscale("log")
 ax.set_xlabel("Storage Power Capacity (GW)")
 ax.set_ylabel("Storage Duration (h)")
 ax.set_yticks([10, 100, 1000])
-ax.set_yticks([2, 3, 4, 5, 6, 7, 8, 9, 20, 30, 40, 50, 60, 70, 80, 90, 200, 300, 400, 500, 600, 700, 800, 900],
-              minor=True)
+ax.set_yticks(
+    [
+        2,
+        3,
+        4,
+        5,
+        6,
+        7,
+        8,
+        9,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        200,
+        300,
+        400,
+        500,
+        600,
+        700,
+        800,
+        900,
+    ],
+    minor=True,
+)
 ax.set_yticklabels(["10", "100", "1000"])
 plt.tight_layout()
 
 # %%
-save_figure("figure-s5-duration-cdf-cost-scenarios.png")
+save_figure("figure-s5-duration-cdf-cost-scenarios.png")

papers/Martin_Staadecker_et_al_2022/figure-s6-duration-cdf-cost-scenarios-ca-only.py renamed to papers/Martin_Staadecker_et_al_2022/figure-x2-duration-cdf-cost-scenarios-ca-only.py

@@ -4,24 +4,28 @@
 
 from papers.Martin_Staadecker_et_al_2022.util import (
     set_style,
-    get_scenario, save_figure
+    get_scenario,
+    save_figure,
 )
 from switch_model.tools.graph.main import GraphTools
 
 # Prepare graph tools
-tools = GraphTools(scenarios=[
-    get_scenario("C21", 0.5),
-    get_scenario("C18", 1),
-    get_scenario("C22", 2),
-    get_scenario("C23", 5),
-    get_scenario("C26", 7),
-    get_scenario("C17", 10),
-    get_scenario("C24", 15),
-    get_scenario("1342", 22.43),
-    get_scenario("C25", 40),
-    get_scenario("C19", 70),
-    get_scenario("C20", 102)
-], set_style=False)
+tools = GraphTools(
+    scenarios=[
+        get_scenario("C21", 0.5),
+        get_scenario("C18", 1),
+        get_scenario("C22", 2),
+        get_scenario("C23", 5),
+        get_scenario("C26", 7),
+        get_scenario("C17", 10),
+        get_scenario("C24", 15),
+        get_scenario("1342", 22.43),
+        get_scenario("C25", 40),
+        get_scenario("C19", 70),
+        get_scenario("C20", 102),
+    ],
+    set_style=False,
+)
 tools.pre_graphing(multi_scenario=True)
 
 set_style()
@@ -34,7 +38,7 @@
 duration = tools.transform.load_zone(duration)
 duration = duration[duration.region == "CA"]
 duration["duration"] = duration["duration"] = (
-        duration["OnlineEnergyCapacityMWh"] / duration["OnlinePowerCapacityMW"]
+    duration["OnlineEnergyCapacityMWh"] / duration["OnlinePowerCapacityMW"]
 )
 
 duration = duration.sort_values("duration")
@@ -46,22 +50,60 @@
     duration_scenario["cuml_power"] = duration_scenario.OnlinePowerCapacityMW.cumsum()
     duration_scenario = duration_scenario.set_index("cuml_power")
     duration_scenario = duration_scenario["duration"]
-    line = ax.plot(duration_scenario.index, duration_scenario, drawstyle="steps", label=scenario_name)
+    line = ax.plot(
+        duration_scenario.index,
+        duration_scenario,
+        drawstyle="steps",
+        label=scenario_name,
+    )
     if float(int(scenario_name)) == scenario_name:
         label = str(int(scenario_name))
     else:
         label = str(scenario_name)
-    labellines.labelLine(line[0], duration_scenario.index.max(), outline_width=2, label=label + "$/KWh", align=False,
-                         fontsize="small")
+    labellines.labelLine(
+        line[0],
+        duration_scenario.index.max(),
+        outline_width=2,
+        label=label + "$/KWh",
+        align=False,
+        fontsize="small",
+    )
 
 ax.set_yscale("log")
 ax.set_xlabel("Storage Power Capacity (GW)")
 ax.set_ylabel("Storage Duration (h)")
 ax.set_yticks([10, 100, 1000])
-ax.set_yticks([2, 3, 4, 5, 6, 7, 8, 9, 20, 30, 40, 50, 60, 70, 80, 90, 200, 300, 400, 500, 600, 700, 800, 900],
-              minor=True)
+ax.set_yticks(
+    [
+        2,
+        3,
+        4,
+        5,
+        6,
+        7,
+        8,
+        9,
+        20,
+        30,
+        40,
+        50,
+        60,
+        70,
+        80,
+        90,
+        200,
+        300,
+        400,
+        500,
+        600,
+        700,
+        800,
+        900,
+    ],
+    minor=True,
+)
 ax.set_yticklabels(["10", "100", "1000"])
 plt.tight_layout()
 
 # %%
-save_figure("figure-s6-duration-cdf-cost-scenarios-ca-only.png")
+save_figure("figure-s6-duration-cdf-cost-scenarios-ca-only.png")