From b4011a1d67fc88a69dff15569f3e1aba962d5192 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9line=20Gie=C3=9Fe?= <celine.giesse@uni-hamburg.de>
Date: Fri, 2 Aug 2024 13:12:38 +0200
Subject: [PATCH] clean up code (sea-ice)
---
Sea-ice.ipynb | 56 ++++++++++++++++++++++-----------------------------
1 file changed, 24 insertions(+), 32 deletions(-)
diff --git a/Sea-ice.ipynb b/Sea-ice.ipynb
index 95b4e7e..1831ca7 100644
--- a/Sea-ice.ipynb
+++ b/Sea-ice.ipynb
@@ -69,7 +69,7 @@
},
{
"cell_type": "code",
- "execution_count": 4,
+ "execution_count": 3,
"id": "6fa849af-5817-47d9-a15a-76d6f1d82b00",
"metadata": {},
"outputs": [],
@@ -78,19 +78,19 @@
"GWLcross_ind = {T: np.empty(50, dtype=int) for T in Tglob}\n",
"GWLcross_ind[\"T0\"] = 50 * [20]\n",
"\n",
- "for Ti in range(1, 3):\n",
+ "for T in range(1, 3):\n",
" for i in range(50):\n",
" j = 20\n",
- " while GSATa_rm[i, j] < Tglob_values[Ti]:\n",
+ " while GSATa_rm[i, j] < Tglob_values[T]:\n",
" j += 1\n",
- " GWLcross_ind[Tglob[Ti]][i] = j\n",
+ " GWLcross_ind[Tglob[T]][i] = j\n",
"\n",
"GWL_ind = {T: np.zeros((50, 20), dtype=int) for T in Tglob}\n",
"\n",
- "for Ti in range(3):\n",
+ "for T in range(3):\n",
" for i in range(50):\n",
- " GWL_ind[Tglob[Ti]][i, :] = np.arange(GWLcross_ind[Tglob[Ti]][i] - 10,\n",
- " GWLcross_ind[Tglob[Ti]][i] + 10)"
+ " GWL_ind[Tglob[T]][i, :] = np.arange(GWLcross_ind[Tglob[T]][i] - 10,\n",
+ " GWLcross_ind[Tglob[T]][i] + 10)"
]
},
{
@@ -103,7 +103,7 @@
},
{
"cell_type": "code",
- "execution_count": 5,
+ "execution_count": 4,
"id": "e99f932a-601d-409f-8fb2-892de7fba8b9",
"metadata": {},
"outputs": [],
@@ -117,8 +117,8 @@
" SIA[i,:] = np.array(xr.open_dataset(file)['sia_nh']).reshape((251,12))\n",
"\n",
"# Create GWL samples of September and March SIA\n",
- "SIA_sep_samples = {Tglob[Ti]: [SIA[i, GWL_ind[Tglob[Ti]][i, :], 8] for i in range(50)] for Ti in range(3)}\n",
- "SIA_mar_samples = {Tglob[Ti]: [SIA[i, GWL_ind[Tglob[Ti]][i, :], 2] for i in range(50)] for Ti in range(3)}"
+ "SIA_sep_samples = {Tglob[T]: [SIA[i, GWL_ind[Tglob[T]][i, :], 8] for i in range(50)] for T in range(3)}\n",
+ "SIA_mar_samples = {Tglob[T]: [SIA[i, GWL_ind[Tglob[T]][i, :], 2] for i in range(50)] for T in range(3)}"
]
},
{
@@ -131,7 +131,7 @@
},
{
"cell_type": "code",
- "execution_count": 6,
+ "execution_count": 5,
"id": "b7d94506-010a-4378-a4c3-59af4914014e",
"metadata": {},
"outputs": [],
@@ -161,19 +161,19 @@
"\n",
"# Bias corrected samples for September and March\n",
"SIA_sep_corr_samples = {\n",
- " Tglob[Ti]: [SIA_sep_corr[i, GWL_ind[Tglob[Ti]][i, :]] for i in range(50)]\n",
- " for Ti in range(3)\n",
+ " Tglob[T]: [SIA_sep_corr[i, GWL_ind[Tglob[T]][i, :]] for i in range(50)]\n",
+ " for T in range(3)\n",
"}\n",
"\n",
"SIA_mar_corr_samples = {\n",
- " Tglob[Ti]: [SIA_mar_corr[i, GWL_ind[Tglob[Ti]][i, :]] for i in range(50)]\n",
- " for Ti in range(3)\n",
+ " Tglob[T]: [SIA_mar_corr[i, GWL_ind[Tglob[T]][i, :]] for i in range(50)]\n",
+ " for T in range(3)\n",
"}"
]
},
{
"cell_type": "code",
- "execution_count": 7,
+ "execution_count": 6,
"id": "ece4a709-832a-4316-ab0e-dfa9699f0b76",
"metadata": {},
"outputs": [],
@@ -204,7 +204,7 @@
},
{
"cell_type": "code",
- "execution_count": 8,
+ "execution_count": 7,
"id": "adc41e36-aefc-41be-a496-4a50f4fe47f3",
"metadata": {},
"outputs": [
@@ -232,22 +232,22 @@
"fig, ax = plt.subplots(1, 3, figsize=(15, 5))\n",
"\n",
"# Loop over the three GWLs\n",
- "for Ti in range(3):\n",
+ "for T in range(3):\n",
" circle_AO = Circle(center, radius_AO, fc=cmap(0.99))\n",
- " ax[Ti].add_patch(circle_AO)\n",
+ " ax[T].add_patch(circle_AO)\n",
" \n",
" # Draw sea ice area circles\n",
" for perc in range(100, -1, -1):\n",
- " radius_SIA = radius_AO * math.sqrt(np.percentile(SIA_sep_corr_detrend_samples[Tglob[Ti]], perc) / AO_area)\n",
+ " radius_SIA = radius_AO * math.sqrt(np.percentile(SIA_sep_corr_detrend_samples[Tglob[T]], perc) / AO_area)\n",
" circle_SIA = Circle(center, radius_SIA, fc=cmap(perc / 100))\n",
- " ax[Ti].add_patch(circle_SIA)\n",
+ " ax[T].add_patch(circle_SIA)\n",
" \n",
" # Add a white circle at the end\n",
" circle_SIA = Circle(center, radius_SIA, fc='white')\n",
- " ax[Ti].add_patch(circle_SIA)\n",
+ " ax[T].add_patch(circle_SIA)\n",
" \n",
- " ax[Ti].set_aspect('equal')\n",
- " ax[Ti].axis('off')\n",
+ " ax[T].set_aspect('equal')\n",
+ " ax[T].axis('off')\n",
"\n",
"# Add a colorbar\n",
"cax = fig.add_axes([0.94, 0.18, 0.015, 0.64])\n",
@@ -260,14 +260,6 @@
"# Show plot\n",
"plt.show()"
]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "id": "0f9fe742-19db-47fb-9b60-c7b6e26bda4f",
- "metadata": {},
- "outputs": [],
- "source": []
}
],
"metadata": {
--
GitLab