{ "cells": [ { "cell_type": "code", "execution_count": 2, "id": "9f4297e3", "metadata": {}, "outputs": [], "source": [ "import os\n", "import sys\n", "\n", "import numpy as np\n", "import matplotlib as mpl\n", "#mpl.use('Agg')\n", "%matplotlib notebook\n", "import matplotlib.pyplot as plt\n", "from matplotlib.cm import get_cmap\n", "from matplotlib.gridspec import GridSpec\n", "from matplotlib.colors import LightSource\n", "from matplotlib.cm import get_cmap\n", "import datetime\n", "from datetime import timedelta\n", "\n", "import cartopy.crs as ccrs\n", "import cartopy.feature as cf\n", "\n", "import xarray as xr\n", "sys.path.insert(1,'/discover/nobackup/cakelle2/GEOS_CF/Apps/cftools/external/cmcrameri')\n", "from cmcrameri import cm" ] }, { "cell_type": "code", "execution_count": 4, "id": "4bde584d", "metadata": {}, "outputs": [], "source": [ "path = \"/discover/nobackup/projects/gmao/geos_cf_dev/psturm/c360_control/holding/geosgcm_kpp/gcc_test.geosgcm_kpp.20190703_0900z.nc4\"\n", "nc = xr.open_dataset(path)" ] }, { "cell_type": "code", "execution_count": 5, "id": "50ec8829", "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "\n", "
<xarray.Dataset>\n",
       "Dimensions:        (lon: 1440, lat: 721, lev: 72, time: 1)\n",
       "Coordinates:\n",
       "  * lon            (lon) float64 -180.0 -179.8 -179.5 ... 179.2 179.5 179.8\n",
       "  * lat            (lat) float64 -90.0 -89.75 -89.5 -89.25 ... 89.5 89.75 90.0\n",
       "  * lev            (lev) float64 1.0 2.0 3.0 4.0 5.0 ... 69.0 70.0 71.0 72.0\n",
       "  * time           (time) datetime64[ns] 2019-07-03T09:00:00\n",
       "Data variables:\n",
       "    KppIntCounts   (time, lev, lat, lon) float32 ...\n",
       "    Met_SUNCOSmid  (time, lat, lon) float32 ...\n",
       "Attributes:\n",
       "    Comment:      NetCDF-4\n",
       "    Contact:      http://gmao.gsfc.nasa.gov\n",
       "    Conventions:  CF\n",
       "    Filename:     geosgcm_kpp\n",
       "    History:      File written by MAPL_PFIO\n",
       "    Institution:  NASA Global Modeling and Assimilation Office\n",
       "    References:   see MAPL documentation\n",
       "    Source:       GEOSgcm-v10.16.0 experiment_id: gcc_test\n",
       "    Title:        GEOS-Chem_v13_c48_test_run
" ], "text/plain": [ "\n", "Dimensions: (lon: 1440, lat: 721, lev: 72, time: 1)\n", "Coordinates:\n", " * lon (lon) float64 -180.0 -179.8 -179.5 ... 179.2 179.5 179.8\n", " * lat (lat) float64 -90.0 -89.75 -89.5 -89.25 ... 89.5 89.75 90.0\n", " * lev (lev) float64 1.0 2.0 3.0 4.0 5.0 ... 69.0 70.0 71.0 72.0\n", " * time (time) datetime64[ns] 2019-07-03T09:00:00\n", "Data variables:\n", " KppIntCounts (time, lev, lat, lon) float32 ...\n", " Met_SUNCOSmid (time, lat, lon) float32 ...\n", "Attributes:\n", " Comment: NetCDF-4\n", " Contact: http://gmao.gsfc.nasa.gov\n", " Conventions: CF\n", " Filename: geosgcm_kpp\n", " History: File written by MAPL_PFIO\n", " Institution: NASA Global Modeling and Assimilation Office\n", " References: see MAPL documentation\n", " Source: GEOSgcm-v10.16.0 experiment_id: gcc_test\n", " Title: GEOS-Chem_v13_c48_test_run" ] }, "execution_count": 5, "metadata": {}, "output_type": "execute_result" } ], "source": [ "nc" ] }, { "cell_type": "code", "execution_count": 19, "id": "8b63008d", "metadata": {}, "outputs": [], "source": [ "maxloc = np.argmax(nc[\"KppIntCounts\"].values)\n", "maxloc= np.unravel_index(maxloc, nc[\"KppIntCounts\"].shape)" ] }, { "cell_type": "code", "execution_count": 20, "id": "351a2bdd", "metadata": {}, "outputs": [ { "data": { "text/plain": [ "101.67232" ] }, "execution_count": 20, "metadata": {}, "output_type": "execute_result" } ], "source": [ "nc[\"KppIntCounts\"].values[maxloc]" ] }, { "cell_type": "code", "execution_count": 26, "id": "1d37578a", "metadata": {}, "outputs": [ { "data": { "text/plain": [ "-0.39239419" ] }, "execution_count": 26, "metadata": {}, "output_type": "execute_result" } ], "source": [ "nc[\"Met_SUNCOSmid\"].values[0, 481, 356]" ] }, { "cell_type": "code", "execution_count": 24, "id": "893f0d60", "metadata": {}, "outputs": [ { "data": { "text/plain": [ "(1, 721, 1440)" ] }, "execution_count": 24, "metadata": {}, "output_type": "execute_result" } ], "source": [ "nc[\"Met_SUNCOSmid\"].shape" ] }, { "cell_type": "code", "execution_count": 25, "id": "ed25f01e", "metadata": {}, "outputs": [ { "data": { "text/plain": [ "(0, 71, 481, 356)" ] }, "execution_count": 25, "metadata": {}, "output_type": "execute_result" } ], "source": [ "maxloc" ] }, { "cell_type": "code", "execution_count": null, "id": "d4eee639", "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "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.9.9" } }, "nbformat": 4, "nbformat_minor": 5 }