update to 2022

singularity-energy · Oct 2, 2024 · 8696881 · 8696881
1 parent fdc01de
commit 8696881
Showing 1 changed file with 61 additions and 14 deletions.
diff --git a/notebooks/explore_data/export_generator_data.ipynb b/notebooks/explore_data/export_generator_data.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 26,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -31,28 +31,68 @@
     "import gross_to_net_generation\n",
     "import eia930\n",
     "\n",
-    "year = 2021\n",
+    "year = 2022\n",
     "path_prefix = f\"{year}/\""
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 27,
    "metadata": {},
    "outputs": [],
    "source": [
     "# load data for each generator\n",
-    "pudl_out = load_data.initialize_pudl_out(year)\n",
-    "gens = pudl_out.gens_eia860()\n",
+    "gens = load_data.load_pudl_table(\n",
+    "    \"generators_eia860\",\n",
+    "    year,\n",
+    "    columns=[\n",
+    "        \"report_date\",\n",
+    "        \"plant_id_eia\",\n",
+    "        \"generator_id\",\n",
+    "        \"prime_mover_code\",\n",
+    "        \"capacity_mw\",\n",
+    "        \"generator_retirement_date\",\n",
+    "        \"operational_status\",\n",
+    "        \"operational_status_code\",\n",
+    "        \"current_planned_generator_operating_date\",\n",
+    "        \"planned_generator_retirement_date\",\n",
+    "    ],\n",
+    ")\n",
     "\n",
-    "# load data about the balancing authority of each plant\n",
-    "plant_ba = pudl_out.plants_eia860()[\n",
-    "    [\"plant_id_eia\", \"balancing_authority_code_eia\"]\n",
-    "].rename(columns={\"balancing_authority_code_eia\": \"ba_code\"})\n",
+    "gens_entity = load_data.load_pudl_table(\n",
+    "    \"generators_entity_eia\",\n",
+    "    columns=[\n",
+    "        \"plant_id_eia\",\n",
+    "        \"generator_id\",\n",
+    "        \"generator_operating_date\",\n",
+    "        \"original_planned_generator_operating_date\",\n",
+    "    ],\n",
+    ")\n",
     "\n",
-    "# merge the ba code into the gens data\n",
-    "gens = gens.merge(plant_ba, how=\"left\", on=\"plant_id_eia\", validate=\"m:1\")\n",
+    "plants_entity = load_data.load_pudl_table(\n",
+    "    \"plants_entity_eia\", columns=[\"plant_id_eia\", \"latitude\", \"longitude\"]\n",
+    ")\n",
+    "\n",
+    "plant_ba = load_data.load_pudl_table(\n",
+    "    \"plants_eia860\", year, columns=[\"plant_id_eia\", \"balancing_authority_code_eia\"]\n",
+    ").rename(columns={\"balancing_authority_code_eia\": \"ba_code\"})\n",
     "\n",
+    "# merge the ba code into the gens data\n",
+    "gens = (\n",
+    "    gens.merge(\n",
+    "        gens_entity, how=\"left\", on=[\"plant_id_eia\", \"generator_id\"], validate=\"m:1\"\n",
+    "    )\n",
+    "    .merge(plant_ba, how=\"left\", on=\"plant_id_eia\", validate=\"m:1\")\n",
+    "    .merge(plants_entity, how=\"left\", on=\"plant_id_eia\", validate=\"m:1\")\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 28,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "# load wind data\n",
     "wind = pd.read_excel(\n",
     "    downloads_folder(f\"eia860/eia860{year}/3_2_Wind_Y{year}.xlsx\"),\n",
@@ -74,8 +114,15 @@
     "wind = wind[[\"plant_id_eia\", \"generator_id\", \"rated_speed_m_per_s\", \"hub_height_m\"]]\n",
     "\n",
     "# merge the wind data into the gens data\n",
-    "gens = gens.merge(wind, how=\"left\", on=[\"plant_id_eia\",\"generator_id\"])\n",
-    "\n",
+    "gens = gens.merge(wind, how=\"left\", on=[\"plant_id_eia\",\"generator_id\"])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 29,
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "gens.to_csv(outputs_folder(f\"gens_{year}.csv\"), index=False)"
    ]
   }
@@ -96,7 +143,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.9"
+   "version": "3.11.4"
   },
   "orig_nbformat": 4,
   "vscode": {