temperature_spatial_meteorology.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c0cf9a64",
   "metadata": {},
   "outputs": [],
   "source": [
    "#===== This code enable to produce total precipitation  from WRF-Chem and satallite observation ==========================\n",
    "import matplotlib.pyplot as plt\n",
    "from mpl_toolkits.basemap import Basemap\n",
    "from netCDF4 import Dataset\n",
    "import numpy as np\n",
    "import cmaps\n",
    "from matplotlib.patches import Patch\n",
    "from pylab import *\n",
    "from wrf import (to_np, getvar, interplevel, smooth2d, get_basemap, latlon_coords, ALL_TIMES)\n",
    "from matplotlib.pyplot import figure\n",
    "import matplotlib.gridspec as gridspec\n",
    "import wrf\n",
    "import scipy as sp\n",
    "import scipy.ndimage\n",
    "from mpl_toolkits.axes_grid1 import make_axes_locatable\n",
    "from matplotlib import rcParams\n",
    "\n",
    "\"\"\"Control font\"\"\"\n",
    "matplotlib.rcParams['font.family'] = \"sans-serif\"\n",
    "matplotlib.rcParams['font.sans-serif'] = \"Times New Roman\"\n",
    "\n",
    "#=======Monsoon-WRF===\n",
    "nc1 = Dataset(\"wrfout_d01_2017-05-01_00_00_00.nc\")\n",
    "nc2 = Dataset(\"wrfout_d01_2017-06-01_00_00_00\")\n",
    "nc3 = Dataset(\"wrfout_d01_2017-07-01_00_00_00\")\n",
    "nc4 = Dataset(\"wrfout_d01_2017-08-01_00_00_00\")\n",
    "nc5 = Dataset(\"wrfout_d01_2017-09-01_00_00_00\")\n",
    "tw1 = getvar(nc1, \"T2\",timeidx=-1) #,timeidx=ALL_TIMES,method=\"join\", squeeze=False)     #ALL_TIMES)\n",
    "tw2 = getvar(nc2, \"T2\",timeidx=-1) #timeidx=ALL_TIMES,method=\"join\", squeeze=False)      #ALL_TIMES)\n",
    "tw3 = getvar(nc3, \"T2\",timeidx=-1) #timeidx=ALL_TIMES,method=\"join\", squeeze=False)     #ALL_TIMES)\n",
    "tw4 = getvar(nc4, \"T2\",timeidx=-1) #timeidx=ALL_TIMES,method=\"join\", squeeze=False)\n",
    "tw5 = getvar(nc5, \"T2\",timeidx=-1) #timeidx=ALL_TIMES,method=\"join\", squeeze=False)\n",
    "tc1 = tw1-273\n",
    "tc2 = tw2-273\n",
    "tc3 = tw3-273\n",
    "tc4 = tw4-273\n",
    "tc5 = tw5-273\n",
    "monsoon_wrf_tc = (tc1 + tc2 + tc3 + tc4 + tc5)/5\n",
    "#===== Non-Monsoon-WRF=======\n",
    "nc1 = Dataset(\"wrfout_d01_2017-03-01_00_00_00.nc\")\n",
    "nc2 = Dataset(\"wrfout_d01_2017-04-01_00_00_00.nc\")\n",
    "nc3 = Dataset(\"wrfout_d01_2017-01-01_00_00_00\")\n",
    "nc4 = Dataset(\"wrfout_d01_2017-02-01_00_00_00\")\n",
    "nc5 = Dataset(\"wrfout_d01_2017-10-01_00_00_00\")\n",
    "nc6 = Dataset(\"wrfout_d01_2017-11-01_00_00_00\")\n",
    "nc7 = Dataset(\"wrfout_d01_2017-12-01_00_00_00\")\n",
    "t1 = getvar(nc1, \"T2\",timeidx=-1)     #ALL_TIMES)\n",
    "t2 = getvar(nc2, \"T2\",timeidx=-1)      #ALL_TIMES)\n",
    "t3 = getvar(nc3, \"T2\",timeidx=-1)     #ALL_TIMES)\n",
    "t4 = getvar(nc4, \"T2\",timeidx=-1)\n",
    "t5 = getvar(nc5, \"T2\",timeidx=-1)\n",
    "t6 = getvar(nc6, \"T2\",timeidx=-1)\n",
    "t7 = getvar(nc7, \"T2\",timeidx=-1)\n",
    "tc1 = t1-273\n",
    "tc2 = t2-273\n",
    "tc3 = t3-273\n",
    "tc4 = t4-273\n",
    "tc5 = t5-273\n",
    "tc6 = t6-273\n",
    "tc7 = t7-273\n",
    "nonmonsoon_wrf_tc = (tc1 + tc2 + tc3 + tc4 + tc5 + tc6 + tc7)/7\n",
    "\n",
    "#=== Monsoon-CRU==\n",
    "cru_mon = \"CRU_Monsoon_temp_new.nc\" \n",
    "data1 = Dataset(cru_mon)\n",
    "temp_mon1 = data1.variables['tmp'][:] \n",
    "monsoon_cru_tc = np.mean(temp_mon1[:,:,:], axis = 0)\n",
    "lon1,lat1 = np.meshgrid(data1.variables['lon'][:], data1.variables['lat'][:])\n",
    "#====Non-Monsoon-CRU====\n",
    "cru_nonmon = \"CRU_Non-Monsoon_temp_new.nc\" \n",
    "data2 = Dataset(cru_nonmon)\n",
    "temp_nonmon2 = data2.variables['tmp'][:]\n",
    "nonmonsoon_cru_tc = np.mean(temp_nonmon2[:,:,:], axis = 0)\n",
    "#=== Monsoon-ERA5===\n",
    "era_mon = \"ERA5-Monsoon_temp_new.nc\" # the location of the data\n",
    "data3 = Dataset(era_mon)\n",
    "temp_mon3 = data3.variables['t2m'][:] # for t\n",
    "temp_av_mon3 = np.mean(temp_mon3[:,:,:], axis = 0)\n",
    "monsoon_era_tc = temp_av_mon3 - 273  # k to deg C\n",
    "lon2,lat2 = np.meshgrid(data3.variables['longitude'][:], data3.variables['latitude'][:])\n",
    "#=== Non-Monsoon-ERA5==\n",
    "era_nonmon = \"ERA5-Non-Monsoon_temp_new.nc\" # the location of the data\n",
    "data4 = Dataset(era_nonmon)\n",
    "temp_nonmon4 = data4.variables['t2m'][:]\n",
    "temp_av_nonmon4 = np.mean(temp_nonmon4[:,:,:], axis = 0)\n",
    "nonmonsoon_era_tc = temp_av_nonmon4 - 273\n",
    "\n",
    "#==== Now Plot======\n",
    "fig,ax = plt.subplots(3,2,figsize=(8,8))\n",
    "\n",
    "#===== WRF-Monsoon====\n",
    "gridspec.GridSpec(3,2)\n",
    "plt.subplot2grid((3,2), (0,0))\n",
    "\n",
    "lats, lons = latlon_coords(monsoon_wrf_tc)\n",
    "#m = Basemap(projection='merc',llcrnrlon=45,llcrnrlat=12,urcrnrlon=118,urcrnrlat=52,resolution='h')\n",
    "m = Basemap(projection='lcc',width=12000000,height=9000000,area_thresh=1000,\\\n",
    "            lat_1=25,lat_2=25,lat_0=25,lon_0=90,llcrnrlon=45,llcrnrlat=0,\\\n",
    "             urcrnrlon=139,urcrnrlat=55,resolution='h')\n",
    "#lon, lat = np.meshgrid(lons, lats)\n",
    "x, y = m(to_np(lons), to_np(lats))\n",
    "ax = m.pcolormesh(x,y,monsoon_wrf_tc,cmap=cmaps.BlueWhiteOrangeRed) #cmaps.WhiteBlueGreenYellowRed)\n",
    "#========== Add Grid lines =======================================================================================\n",
    "m.drawparallels(np.arange(10, 60, 15), linewidth=0.5, dashes=[4, 1], labels=[1,0,0,0], fontsize=10, color='black')\n",
    "m.drawmeridians(np.arange(45, 125,15),linewidth=0.5, dashes=[4, 1],  color='black')\n",
    "#========= Add countries boundasy =============================================================================\n",
    "m.drawcountries()\n",
    "m.drawcoastlines(linewidth=0.3)\n",
    "plt.clim(-20,35)\n",
    "\n",
    "plt.title('Monsoon [May-Sep]',fontsize=12)\n",
    "plt.ylabel('WRF-Chem', labelpad=35,fontsize=12)\n",
    "\n",
    "#========== WRF-Nonmonsoon ===============================\n",
    "plt.subplot2grid((3,2), (0,1))\n",
    "#m = Basemap(projection='merc',llcrnrlon=45,llcrnrlat=12,urcrnrlon=118,urcrnrlat=52,resolution='h')\n",
    "m = Basemap(projection='lcc',width=12000000,height=9000000,area_thresh=1000,\\\n",
    "            lat_1=25,lat_2=25,lat_0=25,lon_0=90,llcrnrlon=45,llcrnrlat=0,\\\n",
    "             urcrnrlon=139,urcrnrlat=55,resolution='h')\n",
    "\n",
    "##lon, lat = np.meshgrid(lons, lats)\n",
    "x, y = m(to_np(lons), to_np(lats))\n",
    "ax = m.pcolormesh(x,y,nonmonsoon_wrf_tc,cmap=cmaps.BlueWhiteOrangeRed) #cmaps.WhiteBlueGreenYellowRed)\n",
    "m.drawparallels(np.arange(10, 60, 14), linewidth=0.5, dashes=[4, 1], color='black')\n",
    "m.drawmeridians(np.arange(45, 125,15),linewidth=0.5, dashes=[4, 1],  color='black')\n",
    "m.drawcountries()\n",
    "m.drawcoastlines(linewidth=0.3)\n",
    "plt.clim(-20,35)\n",
    "plt.title('Non-monsoon [Oct-April]',fontsize=12)\n",
    "\n",
    "#==== CRU- Monsoon ========\n",
    "plt.subplot2grid((3,2), (1,0))\n",
    "#m = Basemap(projection='merc',llcrnrlon=45,llcrnrlat=12,urcrnrlon=118,urcrnrlat=52,resolution='h')\n",
    "m = Basemap(projection='lcc',width=12000000,height=9000000,area_thresh=1000,\\\n",
    "            lat_1=25,lat_2=25,lat_0=25,lon_0=90,llcrnrlon=45,llcrnrlat=0,\\\n",
    "             urcrnrlon=139,urcrnrlat=55,resolution='h')\n",
    "\n",
    "x, y = m(lon1, lat1)\n",
    "ax = m.pcolormesh(x,y,np.squeeze(monsoon_cru_tc),cmap=cmaps.BlueWhiteOrangeRed) \n",
    "m.drawparallels(np.arange(10, 60, 15), linewidth=0.5, dashes=[4, 1], labels=[1,0,0,0], fontsize=10, color='black')\n",
    "m.drawmeridians(np.arange(45, 125,15),linewidth=0.5, dashes=[4, 1],  color='black')\n",
    "m.drawcountries()\n",
    "m.drawcoastlines(linewidth=0.3)\n",
    "plt.clim(-20,35)\n",
    "plt.ylabel('CRU', labelpad=35,fontsize=12)\n",
    "\n",
    "#==== CRU- Non-Monsoon ========\n",
    "plt.subplot2grid((3,2), (1,1))\n",
    "#m = Basemap(projection='merc',llcrnrlon=45,llcrnrlat=12,urcrnrlon=118,urcrnrlat=52,resolution='h')\n",
    "m = Basemap(projection='lcc',width=12000000,height=9000000,area_thresh=1000,\\\n",
    "            lat_1=25,lat_2=25,lat_0=25,lon_0=90,llcrnrlon=45,llcrnrlat=0,\\\n",
    "             urcrnrlon=139,urcrnrlat=55,resolution='h')\n",
    "\n",
    "x, y = m(lon1, lat1)\n",
    "ax = m.pcolormesh(x,y,np.squeeze(nonmonsoon_cru_tc),cmap=cmaps.BlueWhiteOrangeRed)\n",
    "m.drawparallels(np.arange(10, 60, 15), linewidth=0.5, dashes=[4, 1], color='black')\n",
    "m.drawmeridians(np.arange(45, 125,15),linewidth=0.5, dashes=[4, 1], color='black')\n",
    "m.drawcountries()\n",
    "m.drawcoastlines(linewidth=0.3)\n",
    "plt.clim(-20,35)\n",
    "\n",
    "#======== ERA5-Monsoon =============================\n",
    "plt.subplot2grid((3,2), (2,0))\n",
    "#m = Basemap(projection='merc',llcrnrlon=45,llcrnrlat=12,urcrnrlon=118,urcrnrlat=52,resolution='h')\n",
    "m = Basemap(projection='lcc',width=12000000,height=9000000,area_thresh=1000,\\\n",
    "            lat_1=25,lat_2=25,lat_0=25,lon_0=90,llcrnrlon=45,llcrnrlat=0,\\\n",
    "             urcrnrlon=139,urcrnrlat=55,resolution='h')\n",
    "x, y = m(lon2, lat2)\n",
    "ax = m.pcolormesh(x,y,np.squeeze(monsoon_era_tc),cmap=cmaps.BlueWhiteOrangeRed)\n",
    "m.drawparallels(np.arange(10, 60, 15), linewidth=0.5, dashes=[4, 1], labels=[1,0,0,0], fontsize=10, color='black')\n",
    "m.drawmeridians(np.arange(45, 125,15),linewidth=0.5, dashes=[4, 1], labels=[0,0,0,1], fontsize=10, color='black')\n",
    "m.drawcountries()\n",
    "m.drawcoastlines(linewidth=0.3)\n",
    "plt.clim(-20,35)\n",
    "plt.ylabel('ERA5', labelpad=35,fontsize=12)\n",
    "#======== ERA5-Non-Monsoon =============================\n",
    "plt.subplot2grid((3,2), (2,1))\n",
    "#m = Basemap(projection='merc',llcrnrlon=45,llcrnrlat=12,urcrnrlon=118,urcrnrlat=52,resolution='h')\n",
    "m = Basemap(projection='lcc',width=12000000,height=9000000,area_thresh=1000,\\\n",
    "            lat_1=25,lat_2=25,lat_0=25,lon_0=90,llcrnrlon=45,llcrnrlat=0,\\\n",
    "             urcrnrlon=139,urcrnrlat=55,resolution='h')\n",
    "x, y = m(lon2, lat2)\n",
    "ax = m.pcolormesh(x,y,np.squeeze(nonmonsoon_era_tc),cmap=cmaps.BlueWhiteOrangeRed)\n",
    "m.drawparallels(np.arange(10, 60, 15), linewidth=0.5, dashes=[4, 1], color='black')\n",
    "m.drawmeridians(np.arange(45, 125,15),linewidth=0.5, dashes=[4, 1], labels=[0,0,0,1], fontsize=10, color='black')\n",
    "m.drawcountries()\n",
    "m.drawcoastlines(linewidth=0.3)\n",
    "plt.clim(-20,35)\n",
    "\n",
    "fig.subplots_adjust(top=0.967,\n",
    "                    bottom=0.11,\n",
    "                    left=0.085,\n",
    "                    right=1,\n",
    "                    hspace=0.025,\n",
    "                    wspace=0.0)\n",
    "# Colorbar\n",
    "cax = fig.add_axes([0.11, 0.06,0.87,0.02])   # left, bottom, width, and height\n",
    "cb = fig.colorbar(ax,  cax=cax, orientation='horizontal')\n",
    "cb.set_label(label='Temperature [°C]',size=10)\n",
    "\n",
    "\n",
    "plt.savefig(\"/mnt/g/2nd_Paper/Temperature.png\",dpi=300)\n",
    "#plt.show()\n",
    "\n",
    "\n"
   ]
  }
 ],
 "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.9.1"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}