{ "cells": [ { "cell_type": "code", "execution_count": 16, "metadata": { "execution": { "iopub.execute_input": "2021-04-07T13:22:43.059942Z", "iopub.status.busy": "2021-04-07T13:22:43.059448Z", "iopub.status.idle": "2021-04-07T13:22:43.067459Z", "shell.execute_reply": "2021-04-07T13:22:43.066451Z", "shell.execute_reply.started": "2021-04-07T13:22:43.059885Z" } }, "outputs": [ { "data": { "text/plain": [ "" ] }, "execution_count": 16, "metadata": {}, "output_type": "execute_result" } ], "source": [ "import xarray as xr\n", "import pandas as pd\n", "\n", "xr.set_options(keep_attrs=True, display_style=\"html\")" ] }, { "cell_type": "code", "execution_count": 15, "metadata": { "execution": { "iopub.execute_input": "2021-04-07T13:21:30.271070Z", "iopub.status.busy": "2021-04-07T13:21:30.270501Z", "iopub.status.idle": "2021-04-07T13:21:30.276883Z", "shell.execute_reply": "2021-04-07T13:21:30.275563Z", "shell.execute_reply.started": "2021-04-07T13:21:30.271003Z" } }, "outputs": [], "source": [ "station = \"Davis\"\n", "inpath = \"/projekt2/hochatm/akuchar/from_GStobel/analysis_h5only/\"" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "# LTC group" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "execution": { "iopub.execute_input": "2021-04-07T13:15:08.326504Z", "iopub.status.busy": "2021-04-07T13:15:08.326020Z", "iopub.status.idle": "2021-04-07T13:15:08.719757Z", "shell.execute_reply": "2021-04-07T13:15:08.718928Z", "shell.execute_reply.started": "2021-04-07T13:15:08.326443Z" } }, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
<xarray.Dataset>\n",
       "Dimensions:             (alt: 26, phony_dim_2: 4, phony_dim_3: 12, phony_dim_4: 1, time: 192)\n",
       "Coordinates:\n",
       "  * alt                 (alt) float64 70.0 72.0 74.0 76.0 ... 116.0 118.0 120.0\n",
       "  * time                (time) datetime64[ns] 2005-01-31 ... 2020-12-31\n",
       "Dimensions without coordinates: phony_dim_2, phony_dim_3, phony_dim_4\n",
       "Data variables:\n",
       "    A12u_mod            (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A12u_mod_err        (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A12u_sol_cycle      (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A12u_sol_cycle_err  (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A12um_mod           (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A12um_mod_err       (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A12v_mod            (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A12v_mod_err        (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A12v_sol_cycle      (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A12v_sol_cycle_err  (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A12vm_mod           (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A12vm_mod_err       (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24u_mod            (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24u_mod_err        (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24u_sol_cycle      (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A24u_sol_cycle_err  (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A24um_mod           (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24um_mod_err       (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24v_mod            (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24v_mod_err        (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24v_sol_cycle      (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A24v_sol_cycle_err  (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    A24vm_mod           (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    A24vm_mod_err       (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    GW_mod              (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    GW_mod_err          (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    PW_mod              (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    PW_mod_err          (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    day                 (time, phony_dim_4) float64 15.0 15.0 15.0 ... 15.0 15.0\n",
       "    month               (time, phony_dim_4) float64 1.0 2.0 3.0 ... 11.0 12.0\n",
       "    p12u_sol_cycle      (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    p12u_sol_cycle_err  (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    p12v_sol_cycle      (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    p12v_sol_cycle_err  (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    p24u_sol_cycle      (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    p24u_sol_cycle_err  (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    p24v_sol_cycle      (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    p24v_sol_cycle_err  (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    pu0_sol_cycle       (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    pu0_sol_cycle_err   (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    pv0_sol_cycle       (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    pv0_sol_cycle_err   (phony_dim_3, alt) float64 nan nan nan ... nan nan nan\n",
       "    season              (phony_dim_3, phony_dim_4) float64 7.378e+05 ... 7.38...\n",
       "    u0_mod              (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    u0_mod_err          (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    u0_sol_cycle        (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    u0_sol_cycle_err    (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    u0m_mod             (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    u0m_mod_err         (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    u0m_sol_cycle       (phony_dim_2, alt) float64 nan nan nan ... nan nan nan\n",
       "    u0m_sol_cycle_err   (phony_dim_2, alt) float64 nan nan nan ... nan nan nan\n",
       "    v0_mod              (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    v0_mod_err          (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    v0_sol_cycle        (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    v0_sol_cycle_err    (phony_dim_2, phony_dim_3, alt) float64 nan nan ... nan\n",
       "    v0m_mod             (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    v0m_mod_err         (time, alt) float64 nan nan nan nan ... nan nan nan nan\n",
       "    v0m_sol_cycle       (phony_dim_2, alt) float64 nan nan nan ... nan nan nan\n",
       "    v0m_sol_cycle_err   (phony_dim_2, alt) float64 nan nan nan ... nan nan nan\n",
       "    year                (time, phony_dim_4) float64 2.005e+03 ... 2.02e+03
" ], "text/plain": [ "\n", "Dimensions: (alt: 26, phony_dim_2: 4, phony_dim_3: 12, phony_dim_4: 1, time: 192)\n", "Coordinates:\n", " * alt (alt) float64 70.0 72.0 74.0 76.0 ... 116.0 118.0 120.0\n", " * time (time) datetime64[ns] 2005-01-31 ... 2020-12-31\n", "Dimensions without coordinates: phony_dim_2, phony_dim_3, phony_dim_4\n", "Data variables:\n", " A12u_mod (time, alt) float64 ...\n", " A12u_mod_err (time, alt) float64 ...\n", " A12u_sol_cycle (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A12u_sol_cycle_err (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A12um_mod (time, alt) float64 ...\n", " A12um_mod_err (time, alt) float64 ...\n", " A12v_mod (time, alt) float64 ...\n", " A12v_mod_err (time, alt) float64 ...\n", " A12v_sol_cycle (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A12v_sol_cycle_err (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A12vm_mod (time, alt) float64 ...\n", " A12vm_mod_err (time, alt) float64 ...\n", " A24u_mod (time, alt) float64 ...\n", " A24u_mod_err (time, alt) float64 ...\n", " A24u_sol_cycle (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A24u_sol_cycle_err (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A24um_mod (time, alt) float64 ...\n", " A24um_mod_err (time, alt) float64 ...\n", " A24v_mod (time, alt) float64 ...\n", " A24v_mod_err (time, alt) float64 ...\n", " A24v_sol_cycle (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A24v_sol_cycle_err (phony_dim_2, phony_dim_3, alt) float64 ...\n", " A24vm_mod (time, alt) float64 ...\n", " A24vm_mod_err (time, alt) float64 ...\n", " GW_mod (time, alt) float64 ...\n", " GW_mod_err (time, alt) float64 ...\n", " PW_mod (time, alt) float64 ...\n", " PW_mod_err (time, alt) float64 ...\n", " day (time, phony_dim_4) float64 ...\n", " month (time, phony_dim_4) float64 1.0 2.0 3.0 ... 11.0 12.0\n", " p12u_sol_cycle (phony_dim_3, alt) float64 ...\n", " p12u_sol_cycle_err (phony_dim_3, alt) float64 ...\n", " p12v_sol_cycle (phony_dim_3, alt) float64 ...\n", " p12v_sol_cycle_err (phony_dim_3, alt) float64 ...\n", " p24u_sol_cycle (phony_dim_3, alt) float64 ...\n", " p24u_sol_cycle_err (phony_dim_3, alt) float64 ...\n", " p24v_sol_cycle (phony_dim_3, alt) float64 ...\n", " p24v_sol_cycle_err (phony_dim_3, alt) float64 ...\n", " pu0_sol_cycle (phony_dim_3, alt) float64 ...\n", " pu0_sol_cycle_err (phony_dim_3, alt) float64 ...\n", " pv0_sol_cycle (phony_dim_3, alt) float64 ...\n", " pv0_sol_cycle_err (phony_dim_3, alt) float64 ...\n", " season (phony_dim_3, phony_dim_4) float64 ...\n", " u0_mod (time, alt) float64 ...\n", " u0_mod_err (time, alt) float64 ...\n", " u0_sol_cycle (phony_dim_2, phony_dim_3, alt) float64 ...\n", " u0_sol_cycle_err (phony_dim_2, phony_dim_3, alt) float64 ...\n", " u0m_mod (time, alt) float64 ...\n", " u0m_mod_err (time, alt) float64 ...\n", " u0m_sol_cycle (phony_dim_2, alt) float64 ...\n", " u0m_sol_cycle_err (phony_dim_2, alt) float64 ...\n", " v0_mod (time, alt) float64 ...\n", " v0_mod_err (time, alt) float64 ...\n", " v0_sol_cycle (phony_dim_2, phony_dim_3, alt) float64 ...\n", " v0_sol_cycle_err (phony_dim_2, phony_dim_3, alt) float64 ...\n", " v0m_mod (time, alt) float64 ...\n", " v0m_mod_err (time, alt) float64 ...\n", " v0m_sol_cycle (phony_dim_2, alt) float64 ...\n", " v0m_sol_cycle_err (phony_dim_2, alt) float64 ...\n", " year (time, phony_dim_4) float64 2.005e+03 ... 2.02e+03" ] }, "execution_count": 4, "metadata": {}, "output_type": "execute_result" } ], "source": [ "ds_info = xr.open_dataset(f\"{inpath}{station}_solar_cycle.h5\", group=\"info\")\n", "ds_info = ds_info.rename({\"phony_dim_7\": \"alt\"})\n", "\n", "ds = xr.open_dataset(f\"{inpath}{station}_solar_cycle.h5\", group=\"LTC\")\n", "s_year = int(ds[\"year\"].values[0][0])\n", "s_month = int(ds[\"month\"].values[0][0])\n", "time_shape = ds[\"year\"].values.shape[0]\n", "\n", "ds = ds.assign_coords(\n", " {\n", " \"alt\": ds_info[\"alt\"].squeeze(),\n", " \"time\": pd.date_range(f\"{s_year}-{s_month}\", freq=\"M\", periods=time_shape),\n", " }\n", ")\n", "ds = ds.rename({\"phony_dim_0\": \"time\", \"phony_dim_1\": \"alt\"})\n", "ds" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "# Wind group" ] }, { "cell_type": "code", "execution_count": 14, "metadata": { "execution": { "iopub.execute_input": "2021-04-07T13:21:05.664020Z", "iopub.status.busy": "2021-04-07T13:21:05.663531Z", "iopub.status.idle": "2021-04-07T13:21:05.698828Z", "shell.execute_reply": "2021-04-07T13:21:05.698059Z", "shell.execute_reply.started": "2021-04-07T13:21:05.663963Z" } }, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
<xarray.Dataset>\n",
       "Dimensions:  (alt: 26, time: 5844)\n",
       "Coordinates:\n",
       "  * alt      (alt) float64 70.0 72.0 74.0 76.0 78.0 ... 114.0 116.0 118.0 120.0\n",
       "  * time     (time) datetime64[ns] 2005-01-01 2005-01-02 ... 2020-12-31\n",
       "Data variables:\n",
       "    u0       (time, alt) float64 ...\n",
       "    u0_err   (time, alt) float64 ...\n",
       "    v0       (time, alt) float64 ...\n",
       "    v0_err   (time, alt) float64 ...\n",
       "    w0       (time, alt) float64 ...\n",
       "    w0_err   (time, alt) float64 ...
" ], "text/plain": [ "\n", "Dimensions: (alt: 26, time: 5844)\n", "Coordinates:\n", " * alt (alt) float64 70.0 72.0 74.0 76.0 78.0 ... 114.0 116.0 118.0 120.0\n", " * time (time) datetime64[ns] 2005-01-01 2005-01-02 ... 2020-12-31\n", "Data variables:\n", " u0 (time, alt) float64 ...\n", " u0_err (time, alt) float64 ...\n", " v0 (time, alt) float64 ...\n", " v0_err (time, alt) float64 ...\n", " w0 (time, alt) float64 ...\n", " w0_err (time, alt) float64 ..." ] }, "execution_count": 14, "metadata": {}, "output_type": "execute_result" } ], "source": [ "ds = xr.open_dataset(f\"{inpath}{station}_solar_cycle.h5\", group=\"wind\")\n", "ds = ds.rename({\"phony_dim_10\": \"time\", \"phony_dim_11\": \"alt\"})\n", "\n", "s_year = int(ds_info[\"year\"].values[0][0])\n", "s_month = int(ds_info[\"month\"].values[0][0])\n", "s_day = int(ds_info[\"day\"].values[0][0])\n", "\n", "ds = ds.assign_coords(\n", " {\n", " \"alt\": ds_info[\"alt\"].squeeze(),\n", " \"time\": pd.date_range(\n", " f\"{s_year}-{s_month}-{s_day}\", freq=\"D\", periods=ds.time.shape[0]\n", " ),\n", " }\n", ")\n", "ds" ] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.6.10" } }, "nbformat": 4, "nbformat_minor": 4 }