feat/add_ensemble_initial_conditions (#4)

- Update notebook (c59b9a2b95ecf8eaba405288c1489f0372553937)

run_AIFS_ENS_v1.ipynb

@@ -58,12 +58,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "id": "a6cdc784-68cb-4e27-b52a-36e82d994e2f",
    "metadata": {},
    "outputs": [],
    "source": [
-    "PARAM_SFC = [\"10u\", \"10v\", \"2d\", \"2t\", \"msl\", \"skt\", \"sp\", \"tcw\", \"lsm\", \"z\", \"slor\", \"sdor\"]\n",
+    "PARAM_SFC = [\"10u\", \"10v\", \"2d\", \"2t\", \"msl\", \"skt\", \"sp\", \"tcw\"]\n",
+    "PARAM_SFC_FC = [\"lsm\", \"z\", \"slor\", \"sdor\"]\n",
     "PARAM_SOIL =[\"sot\"]\n",
     "PARAM_PL = [\"gh\", \"t\", \"u\", \"v\", \"w\", \"q\"]\n",
     "LEVELS = [1000, 925, 850, 700, 600, 500, 400, 300, 250, 200, 150, 100, 50]\n",
@@ -107,16 +108,20 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "id": "062a8a99-ddb0-43e9-aa28-5b1f926ed8a2",
    "metadata": {},
    "outputs": [],
    "source": [
-    "def get_open_data(param, levelist=[]):\n",
+    "def get_open_data(param, levelist=[], number = None):\n",
     "    fields = defaultdict(list)\n",
     "    # Get the data for the current date and the previous date\n",
     "    for date in [DATE - datetime.timedelta(hours=6), DATE]:\n",
-    "        data = ekd.from_source(\"ecmwf-open-data\", date=date, param=param, levelist=levelist)\n",
+    "        if number is None:\n",
+    "            data = ekd.from_source(\"ecmwf-open-data\", date=date, param=param, levelist=levelist)\n",
+    "        else:\n",
+    "            data = ekd.from_source(\"ecmwf-open-data\", date=date, param=param, levelist=levelist, number=[number], stream='enfo')\n",
+    "        \n",
     "        for f in data:\n",
     "            # Open data is between -180 and 180, we need to shift it to 0-360\n",
     "            assert f.to_numpy().shape == (721,1440)\n",
@@ -139,7 +144,11 @@
    "id": "6d2b0711-f477-48a6-afaf-99fab48b7024",
    "metadata": {},
    "source": [
-    "## Get Input Fields"
+    "## Get Input Fields\n",
+    "\n",
+    "> Here we make a single forecast, from the control initial condition of the ensemble. In operations, this model is run 51 times, once from the control and 50 more times from slightly different initial conditions. \n",
+    "\n",
+    "To get different ensemble initial conditions from the ifs pertubation, set `number` to an ensemble number, between 1-50.\n"
    ]
   },
   {
@@ -149,7 +158,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "fields = {}"
+    "fields = {}\n",
+    "number = None"
    ]
   },
   {
@@ -162,7 +172,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "id": "26b43fc6-b41a-4626-8907-01fa2b3231a9",
    "metadata": {},
    "outputs": [
@@ -175,7 +185,8 @@
     }
    ],
    "source": [
-    "fields.update(get_open_data(param=PARAM_SFC))"
+    "fields.update(get_open_data(param=PARAM_SFC, number=number))\n",
+    "fields.update(get_open_data(param=PARAM_SFC_FC)) # Add constant surface fields, retrieved from fc"
    ]
   },
   {
@@ -193,7 +204,7 @@
     }
    ],
    "source": [
-    "soil=get_open_data(param=PARAM_SOIL,levelist=SOIL_LEVELS)"
+    "soil=get_open_data(param=PARAM_SOIL,levelist=SOIL_LEVELS, number=number)"
    ]
   },
   {
@@ -240,7 +251,7 @@
     }
    ],
    "source": [
-    "fields.update(get_open_data(param=PARAM_PL, levelist=LEVELS))"
+    "fields.update(get_open_data(param=PARAM_PL, levelist=LEVELS, number=number))"
    ]
   },
   {