Skip to content

Commit 914d3ab

Browse files
authored
More verbose variable names (#21)
* use verbose variable names * some breathing space
1 parent cfe6be0 commit 914d3ab

File tree

1 file changed

+17
-15
lines changed

1 file changed

+17
-15
lines changed

β€ŽRecipes/AUS2200_xarray_rolling.ipynbβ€Ž

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1186,7 +1186,7 @@
11861186
"metadata": {},
11871187
"outputs": [],
11881188
"source": [
1189-
"def load_aus2200_variable(vnames, t1, t2, exp_id, lon_slice, lat_slice, freq, hgt_slice=None, chunks=\"auto\"):\n",
1189+
"def load_aus2200_variable(vnames, start_time, end_time, exp_id, lon_slice, lat_slice, freq, hgt_slice=None, chunks=\"auto\"):\n",
11901190
"\n",
11911191
" '''\n",
11921192
" Load variables from the mjo-enso AUS2200 experiment, stored on the ua8 project\n",
@@ -1195,9 +1195,9 @@
11951195
"\n",
11961196
" * vnames: list of names of aus2200 variables\n",
11971197
"\n",
1198-
" * t1: start time in %Y-%m-%d %H:%M\"\n",
1198+
" * start_time: start time in %Y-%m-%d %H:%M\"\n",
11991199
"\n",
1200-
" * t2: start time in %Y-%m-%d %H:%M\"\n",
1200+
" * end_time: end time in %Y-%m-%d %H:%M\"\n",
12011201
" \n",
12021202
" * exp_id: string describing the experiment. either 'mjo-elnino', 'mjo-lanina' or 'mjo-neutral'\n",
12031203
"\n",
@@ -1219,10 +1219,10 @@
12191219
" #We are loading a list of files from disk using xr.open_mfdataset. This preprocessing \n",
12201220
" # just slices the lats, lons and levels we are interested in for each file, which is more efficient\n",
12211221
" def _preprocess(ds):\n",
1222-
" ds = ds.sel(lat=lat_slice,lon=lon_slice)\n",
1222+
" ds = ds.sel(lat=lat_slice, lon=lon_slice)\n",
12231223
" return ds\n",
12241224
" def _preprocess_hgt(ds):\n",
1225-
" ds = ds.sel(lat=lat_slice,lon=lon_slice,lev=hgt_slice)\n",
1225+
" ds = ds.sel(lat=lat_slice, lon=lon_slice, lev=hgt_slice)\n",
12261226
" return ds \n",
12271227
" \n",
12281228
" out = []\n",
@@ -1233,33 +1233,35 @@
12331233
" chunks=chunks, \n",
12341234
" parallel=True, \n",
12351235
" preprocess=_preprocess_hgt, \n",
1236-
" engine=\"h5netcdf\").sel(time=slice(t1,t2))\n",
1236+
" engine=\"h5netcdf\").sel(time=slice(start_time, end_time))\n",
12371237
" else:\n",
12381238
" ds = xr.open_mfdataset(fnames,\n",
12391239
" chunks=chunks,\n",
12401240
" parallel=True,\n",
12411241
" preprocess=_preprocess,\n",
1242-
" engine=\"h5netcdf\").sel(time=slice(t1,t2))\n",
1242+
" engine=\"h5netcdf\").sel(time=slice(start_time, end_time))\n",
12431243
" out.append(ds[vname])\n",
12441244
" \n",
12451245
" return out\n",
12461246
"\n",
12471247
"#Define lat lon slices, equivalent to almost the entire AUS2200 domain\n",
1248-
"lon_slice = slice(108,159)\n",
1249-
"lat_slice = slice(-45.7,-6.831799) \n",
1248+
"lon_slice = slice(108, 159)\n",
1249+
"lat_slice = slice(-45.7, -6.831799) \n",
12501250
"\n",
12511251
"#Define times to slice\n",
1252-
"t1=\"2016-01-01 00:00\"\n",
1253-
"t2=\"2016-02-01 00:00\"\n",
1252+
"start_time=\"2016-01-01 00:00\"\n",
1253+
"end_time=\"2016-02-01 00:00\"\n",
12541254
"\n",
12551255
"#Load air temperature for a single model level (111.7 m)\n",
12561256
"aus2200_ta = load_aus2200_variable([\"ta\"],\n",
1257-
" t1,t2,\n",
1257+
" start_time,\n",
1258+
" end_time,\n",
12581259
" \"mjo-elnino\",\n",
1259-
" lon_slice,lat_slice,\n",
1260+
" lon_slice,\n",
1261+
" lat_slice,\n",
12601262
" \"1hr\",\n",
1261-
" hgt_slice=slice(100,120),\n",
1262-
" chunks=({\"time\":6,\"lat\":-1,\"lon\":-1}) \n",
1263+
" hgt_slice=slice(100, 120),\n",
1264+
" chunks=({\"time\":6, \"lat\":-1, \"lon\":-1}) \n",
12631265
" )[0]\n"
12641266
]
12651267
},

0 commit comments

Comments
Β (0)