Plot_irrigated_fraction_and_irrigation_water_withdrawal_timeseries.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "88e8fc48",
   "metadata": {},
   "outputs": [],
   "source": [
    "%matplotlib inline       \n",
    "import scipy.io as scio\n",
    "import matplotlib.pyplot as plt\n",
    "import matplotlib as mpl\n",
    "import cartopy.feature as cfeature\n",
    "import cartopy.crs as ccrs\n",
    "import netCDF4 as nc\n",
    "import numpy as np\n",
    "from Load_data import Data_from_nc\n",
    "import xarray as xr\n",
    "import math\n",
    "import gc\n",
    " #Data from /theia/scratch/databases/climate/cesm/inputdata/lnd/clm2/surfdata_map/IRRMIP....m\n",
    "def get_data_from_nc(file, variable):\n",
    "    file_obj = nc.Dataset(file)\n",
    "    data = file_obj.variables[variable]\n",
    "    var_data = np.array(data)\n",
    "    var_data = var_data[:, 29:, :]\n",
    "    var_data[var_data > 1000000] = np.nan\n",
    "    var_data = np.squeeze(var_data)\n",
    "    return var_data\n",
    "\n",
    "def get_data_from_mat(file, variable):\n",
    "    var_dict = scio.loadmat(file)\n",
    "    var = var_dict[variable]\n",
    "    var = var[:, 29:] \n",
    "    return var\n",
    "\n",
    "def get_data_from_mat_1(file, variable):\n",
    "    var_dict = scio.loadmat(file)\n",
    "    var = var_dict[variable]\n",
    "    return var\n",
    "\n",
    "def get_data_from_mat_for_calcu(file, variable):\n",
    "    var_dict = scio.loadmat(file)\n",
    "    var = var_dict[variable]\n",
    "    var = var[:, 29:]\n",
    "    return var.T\n",
    "\n",
    "\n",
    "AREA = get_data_from_mat('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/AREA.mat', 'AREA') # Area for CLM_data\n",
    "\n",
    "def calcu_global_IWW(str, AREA, var):\n",
    "    IWW_1901_2014 = get_data_from_nc(str, var)\n",
    "    IWW_globe_1901_2014 = IWW_1901_2014 * (AREA.T) / 1000000\n",
    "    IWW_globe_1901_2014 = np.nansum(IWW_globe_1901_2014, axis=1)\n",
    "    IWW_globe_1901_2014 = np.nansum(IWW_globe_1901_2014, axis=1)\n",
    "    return IWW_globe_1901_2014\n",
    "\n",
    "def calcu_global_IWW_region(str, AREA,region_id, var):\n",
    "    IWW_1901_2014 = get_data_from_nc(str, var)\n",
    "    IWW_globe_1901_2014 = IWW_1901_2014 * (AREA.T) / 1000000\n",
    "    IWW_globe_1901_2014[:, ar6_region!=region_id] = np.nan\n",
    "    IWW_globe_1901_2014 = np.nansum(IWW_globe_1901_2014, axis=1)\n",
    "    IWW_globe_1901_2014 = np.nansum(IWW_globe_1901_2014, axis=1)\n",
    "    return IWW_globe_1901_2014\n",
    "\n",
    "def calcu_global_IWW_region_no_str(IWW_1901_2014, AREA,region_id, var):\n",
    "    IWW_globe_1901_2014 = IWW_1901_2014 * (AREA.T) / 1000000\n",
    "    IWW_globe_1901_2014[:, ar6_region!=region_id] = np.nan\n",
    "    IWW_globe_1901_2014 = np.nansum(IWW_globe_1901_2014, axis=1)\n",
    "    IWW_globe_1901_2014 = np.nansum(IWW_globe_1901_2014, axis=1)\n",
    "    return IWW_globe_1901_2014\n",
    "\n",
    "def calcu_global_AREA(global_irri_land_map):\n",
    "    global_irri_land_map = np.nansum(global_irri_land_map, axis=1)\n",
    "    global_irri_land = np.nansum(global_irri_land_map, axis=1)\n",
    "    return global_irri_land\n",
    "\n",
    "def calcu_global_AREA_region(global_irri_land_map, region_id):\n",
    "    global_irri_land_map[:, ar6_region!=region_id] = np.nan\n",
    "    global_irri_land_map = np.nansum(global_irri_land_map, axis=1)\n",
    "    global_irri_land = np.nansum(global_irri_land_map, axis=1)\n",
    "    return global_irri_land\n",
    "\n",
    "ar6_region = get_data_from_mat_for_calcu('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/ar6_region.mat','ar6_region')\n",
    "\n",
    "irr_diff = get_data_from_mat_for_calcu('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/irr_diff_out.mat', 'irr_diff_out')\n",
    "\n",
    "\n",
    "data_surface = Data_from_nc('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/surfdata_irrigation_method.nc')   #load the data\n",
    "data_irrigation_method = data_surface.load_variable('irrigation_method')\n",
    "data_lon = data_surface.load_variable('LONGXY')\n",
    "data_lon = data_lon[0,:]\n",
    "data_lat = data_surface.load_variable('LATIXY')\n",
    "data_lat = data_lat[29:,0]\n",
    "data_lat = data_lat[:]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5f1d53b2",
   "metadata": {},
   "outputs": [],
   "source": [
    "globe_irr_land_IRR = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land.mat', 'global_irri_land')\n",
    "globe_irr_land_NOI = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_noirr.mat', 'global_irri_land')\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_WNA = calcu_global_AREA_region(global_irri_land_map, 4)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_CNA = calcu_global_AREA_region(global_irri_land_map, 5)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_NCA = calcu_global_AREA_region(global_irri_land_map, 7)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_MED = calcu_global_AREA_region(global_irri_land_map, 20)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_WCA = calcu_global_AREA_region(global_irri_land_map, 33)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_EAS = calcu_global_AREA_region(global_irri_land_map, 36)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_SAS = calcu_global_AREA_region(global_irri_land_map, 38)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n",
    "\n",
    "global_irri_land_SEA = calcu_global_AREA_region(global_irri_land_map, 39)\n",
    "\n",
    "global_irri_land_map = get_data_from_mat_1('/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/input_data/global_irri_land_map.mat', 'global_irri_land_map')\n",
    "\n",
    "global_irri_land_map = global_irri_land_map.transpose(2,1,0)\n",
    "\n",
    "global_irri_land_map = global_irri_land_map[:,29:,:]\n",
    "\n",
    "global_irri_land = calcu_global_AREA(global_irri_land_map)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d0b8d145",
   "metadata": {},
   "outputs": [],
   "source": [
    "str_IWW_CESM2 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_IRR_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2 = calcu_global_IWW(str_IWW_CESM2, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_NOI = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_NOI_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2 = calcu_global_IWW(str_IWW_CESM2_NOI, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_CESM2_IRR01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_IRR01_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2_IRR01 = calcu_global_IWW(str_IWW_CESM2_IRR01, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_NOI01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_NOI01_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2_IRR01 = calcu_global_IWW(str_IWW_CESM2_NOI01, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_CESM2_IRR02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_IRR02_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2_IRR02 = calcu_global_IWW(str_IWW_CESM2_IRR02, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_NOI02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_NOI02_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2_IRR02 = calcu_global_IWW(str_IWW_CESM2_NOI02, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_CESM2_IRR03 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_IRR03_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2_IRR03 = calcu_global_IWW(str_IWW_CESM2_IRR03, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_NOI03 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2/QIRRIG/CESM2_NOI03_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2_IRR03 = calcu_global_IWW(str_IWW_CESM2_NOI03, AREA, 'QIRRIG')\n",
    "\n",
    "IWW_IRR_1901_2014_CESM2 = (IWW_IRR_1901_2014_CESM2_IRR01 + IWW_IRR_1901_2014_CESM2_IRR02 + IWW_IRR_1901_2014_CESM2_IRR03)/3\n",
    "IWW_NOI_1901_2014_CESM2 = (IWW_NOI_1901_2014_CESM2_IRR01 + IWW_NOI_1901_2014_CESM2_IRR02 + IWW_NOI_1901_2014_CESM2_IRR03)/3\n",
    "\n",
    "\n",
    "str_IWW_CESM2_gw = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_IRR_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2_gw = calcu_global_IWW(str_IWW_CESM2_gw, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_gw_NOI = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_NOI_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2_gw = calcu_global_IWW(str_IWW_CESM2_gw_NOI, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_CESM2_gw_IRR01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_IRR01_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2_gw_IRR01 = calcu_global_IWW(str_IWW_CESM2_gw_IRR01, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_gw_NOI01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_NOI01_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2_gw_IRR01 = calcu_global_IWW(str_IWW_CESM2_gw_NOI01, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_CESM2_gw_IRR02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_IRR02_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2_gw_IRR02 = calcu_global_IWW(str_IWW_CESM2_gw_IRR02, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_gw_NOI02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_NOI02_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2_gw_IRR02 = calcu_global_IWW(str_IWW_CESM2_gw_NOI02, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_CESM2_gw_IRR03 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_IRR03_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_CESM2_gw_IRR03 = calcu_global_IWW(str_IWW_CESM2_gw_IRR03, AREA, 'QIRRIG')\n",
    "str_IWW_CESM2_gw_NOI03 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/CESM2_gw/QIRRIG/CESM2_gw_NOI03_YEARLYQIRRIG_timmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_CESM2_gw_IRR03 = calcu_global_IWW(str_IWW_CESM2_gw_NOI03, AREA, 'QIRRIG')\n",
    "\n",
    "IWW_IRR_1901_2014_CESM2_gw = (IWW_IRR_1901_2014_CESM2_gw_IRR01 + IWW_IRR_1901_2014_CESM2_gw_IRR02 + IWW_IRR_1901_2014_CESM2_gw_IRR03)/3\n",
    "IWW_NOI_1901_2014_CESM2_gw = (IWW_NOI_1901_2014_CESM2_gw_IRR01 + IWW_NOI_1901_2014_CESM2_gw_IRR02 + IWW_NOI_1901_2014_CESM2_gw_IRR03)/3\n",
    "\n",
    "str_IWW_NorESM = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_IRR_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_NorESM = calcu_global_IWW(str_IWW_NorESM, AREA, 'QIRRIG')\n",
    "str_IWW_NorESM_NOI = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_NOI_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_NorESM = calcu_global_IWW(str_IWW_NorESM_NOI, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_NorESM_IRR01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_IRR01_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_NorESM_IRR01 = calcu_global_IWW(str_IWW_NorESM_IRR01, AREA, 'QIRRIG')\n",
    "str_IWW_NorESM_NOI01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_NOI01_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_NorESM_IRR01 = calcu_global_IWW(str_IWW_NorESM_NOI01, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_NorESM_IRR02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_IRR02_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_NorESM_IRR02 = calcu_global_IWW(str_IWW_NorESM_IRR02, AREA, 'QIRRIG')\n",
    "str_IWW_NorESM_NOI02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_NOI02_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_NorESM_IRR02 = calcu_global_IWW(str_IWW_NorESM_NOI02, AREA, 'QIRRIG')\n",
    "\n",
    "str_IWW_NorESM_IRR03 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_IRR03_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_IRR_1901_2014_NorESM_IRR03 = calcu_global_IWW(str_IWW_NorESM_IRR03, AREA, 'QIRRIG')\n",
    "str_IWW_NorESM_NOI03 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/after_process/NorESM/QIRRIG/NorESM_NOI03_YEARLYQIRRIG_yearmean_1901_2014.nc'\n",
    "IWW_NOI_1901_2014_NorESM_IRR03 = calcu_global_IWW(str_IWW_NorESM_NOI03, AREA, 'QIRRIG')\n",
    "\n",
    "IWW_IRR_1901_2014_NorESM = (IWW_IRR_1901_2014_NorESM_IRR01 + IWW_IRR_1901_2014_NorESM_IRR02 + IWW_IRR_1901_2014_NorESM_IRR03)/3\n",
    "IWW_NOI_1901_2014_NorESM = (IWW_IRR_1901_2014_NorESM_IRR01 + IWW_IRR_1901_2014_NorESM_IRR02 + IWW_IRR_1901_2014_NorESM_IRR03)/3\n",
    "\n",
    "str_IWW_IPSL = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/IPSL-CM6/IRR01/after_processing/IRR01_1901_2014_irr_Day_Month.nc_YEARLY_0.9x1.25'\n",
    "IWW_IRR_1901_2014_IPSL = calcu_global_IWW(str_IWW_IPSL, AREA, 'irr')\n",
    "\n",
    "str_IWW_IPSL_IRR01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/IPSL-CM6/IRR01/after_processing/IRR01_1901_2014_irr_Day_Month.nc_YEARLY_0.9x1.25'\n",
    "IWW_IRR_1901_2014_IPSL_IRR01 = calcu_global_IWW(str_IWW_IPSL_IRR01, AREA, 'irr')\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "str_IWW_CNRM = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/CNRM/CNRM-CM6-1_hist-irr_r1i1p1f2/after_processing/airrww_mon_CNRM-CM6-1_hist-irr_r1i1p1f2_gr_190101-201412.nc_yearmean_YEARLY_0.9x1.25'\n",
    "IWW_IRR_1901_2014_CNRM = calcu_global_IWW(str_IWW_CNRM, AREA, 'airrww')\n",
    "str_IWW_CNRM_NOI = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/CNRM/CNRM-CM6-1_hist-noirr_r1i1p1f2/after_processing/airrww_mon_CNRM-CM6-1_hist-noirr_r1i1p1f2_gr_190101-201412.nc_yearmean_YEARLY_0.9x1.25'\n",
    "IWW_NOI_1901_2014_CNRM = calcu_global_IWW(str_IWW_CNRM_NOI, AREA, 'airrww')\n",
    "\n",
    "str_IWW_CNRM_IRR01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/CNRM/CNRM-CM6-1_hist-irr_r1i1p1f2/after_processing/airrww_mon_CNRM-CM6-1_hist-irr_r1i1p1f2_gr_190101-201412.nc_yearmean_YEARLY_0.9x1.25'\n",
    "IWW_IRR_1901_2014_CNRM_IRR01 = calcu_global_IWW(str_IWW_CNRM_IRR01, AREA, 'airrww')\n",
    "str_IWW_CNRM_NOI01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/CNRM/CNRM-CM6-1_hist-noirr_r1i1p1f2/after_processing/airrww_mon_CNRM-CM6-1_hist-noirr_r1i1p1f2_gr_190101-201412.nc_yearmean_YEARLY_0.9x1.25'\n",
    "IWW_NOI_1901_2014_CNRM_NOI01 = calcu_global_IWW(str_IWW_CNRM_NOI01, AREA, 'airrww')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5f470b47",
   "metadata": {},
   "outputs": [],
   "source": [
    "str_IWW_E3SM = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/E3SM_cdo/Water_Energy/IRR_QIRRIG_REAL_yearmean_1901_2014.nc_0.9x1.25_YEARLY_yearmean'\n",
    "IWW_IRR_1901_2014_E3SM = calcu_global_IWW(str_IWW_E3SM, AREA, 'QIRRIG_REAL')\n",
    "str_IWW_E3SM_NOI = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/E3SM_cdo/Water_Energy/NOI_QIRRIG_REAL_yearmean_1901_2014.nc_0.9x1.25_YEARLY_yearmean'\n",
    "IWW_NOI_1901_2014_E3SM = calcu_global_IWW(str_IWW_E3SM_NOI, AREA, 'QIRRIG_REAL')\n",
    "\n",
    "str_IWW_E3SM_IRR01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/E3SM_cdo/IRR01/QIRRIG_REAL/IRR01_QIRRIG_REAL_yearmean_1901_2014.nc_0.9x1.25_YEARLY_yearmean'\n",
    "IWW_IRR_1901_2014_E3SM_IRR01 = calcu_global_IWW(str_IWW_E3SM_IRR01, AREA, 'QIRRIG_REAL')\n",
    "str_IWW_E3SM_IRR02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/E3SM_cdo/IRR02/QIRRIG_REAL/IRR02_QIRRIG_REAL_yearmean_1901_2014.nc_0.9x1.25_YEARLY_yearmean'\n",
    "IWW_IRR_1901_2014_E3SM_IRR02 = calcu_global_IWW(str_IWW_E3SM_IRR02, AREA, 'QIRRIG_REAL')\n",
    "str_IWW_E3SM_NOI01 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/E3SM_cdo/NOI01/QIRRIG_REAL/NOI01_QIRRIG_REAL_yearmean_1901_2014.nc_0.9x1.25_YEARLY_yearmean'\n",
    "IWW_IRR_1901_2014_E3SM_NOI01 = calcu_global_IWW(str_IWW_E3SM_NOI01, AREA, 'QIRRIG_REAL')\n",
    "str_IWW_E3SM_NOI02 = '/dodrio/scratch/projects/2022_200/project_output/cesm/yi_yao_IRRMIP/E3SM_cdo/NOI02/QIRRIG_REAL/NOI02_QIRRIG_REAL_yearmean_1901_2014.nc_0.9x1.25_YEARLY_yearmean'\n",
    "IWW_IRR_1901_2014_E3SM_NOI02 = calcu_global_IWW(str_IWW_E3SM_NOI02, AREA, 'QIRRIG_REAL')\n",
    "\n",
    "IWW_IRR_1901_2014_E3SM = (IWW_IRR_1901_2014_E3SM_IRR01 + IWW_IRR_1901_2014_E3SM_IRR02)/2\n",
    "IWW_NOI_1901_2014_E3SM = (IWW_IRR_1901_2014_E3SM_NOI01 + IWW_IRR_1901_2014_E3SM_NOI02)/2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a64959cb",
   "metadata": {},
   "outputs": [],
   "source": [
    "IWW_IRR_1901_2014_CESM2_IRR01 = IWW_IRR_1901_2014_CESM2_IRR01[np.newaxis, :]\n",
    "IWW_IRR_1901_2014_CESM2_IRR02 = IWW_IRR_1901_2014_CESM2_IRR02[np.newaxis, :]\n",
    "IWW_IRR_1901_2014_CESM2_IRR03 = IWW_IRR_1901_2014_CESM2_IRR03[np.newaxis, :]\n",
    "\n",
    "IWW_NOI_1901_2014_CESM2_IRR01 = IWW_NOI_1901_2014_CESM2_IRR01[np.newaxis, :]\n",
    "IWW_NOI_1901_2014_CESM2_IRR02 = IWW_NOI_1901_2014_CESM2_IRR02[np.newaxis, :]\n",
    "IWW_NOI_1901_2014_CESM2_IRR03 = IWW_NOI_1901_2014_CESM2_IRR03[np.newaxis, :]\n",
    "\n",
    "\n",
    "IWW_IRR_1901_2014_CESM2_gw_IRR01 = IWW_IRR_1901_2014_CESM2_gw_IRR01[np.newaxis, :]\n",
    "IWW_IRR_1901_2014_CESM2_gw_IRR02 = IWW_IRR_1901_2014_CESM2_gw_IRR02[np.newaxis, :]\n",
    "IWW_IRR_1901_2014_CESM2_gw_IRR03 = IWW_IRR_1901_2014_CESM2_gw_IRR03[np.newaxis, :]\n",
    "\n",
    "IWW_NOI_1901_2014_CESM2_gw_IRR01 = IWW_NOI_1901_2014_CESM2_gw_IRR01[np.newaxis, :]\n",
    "IWW_NOI_1901_2014_CESM2_gw_IRR02 = IWW_NOI_1901_2014_CESM2_gw_IRR02[np.newaxis, :]\n",
    "IWW_NOI_1901_2014_CESM2_gw_IRR03 = IWW_NOI_1901_2014_CESM2_gw_IRR03[np.newaxis, :]\n",
    "\n",
    "\n",
    "IWW_IRR_1901_2014_NorESM_IRR01 = IWW_IRR_1901_2014_NorESM_IRR01[np.newaxis, :]\n",
    "IWW_IRR_1901_2014_NorESM_IRR02 = IWW_IRR_1901_2014_NorESM_IRR02[np.newaxis, :]\n",
    "IWW_IRR_1901_2014_NorESM_IRR03 = IWW_IRR_1901_2014_NorESM_IRR03[np.newaxis, :]\n",
    "\n",
    "IWW_NOI_1901_2014_NorESM_IRR01 = IWW_NOI_1901_2014_NorESM_IRR01[np.newaxis, :]\n",
    "IWW_NOI_1901_2014_NorESM_IRR02 = IWW_NOI_1901_2014_NorESM_IRR02[np.newaxis, :]\n",
    "IWW_NOI_1901_2014_NorESM_IRR03 = IWW_NOI_1901_2014_NorESM_IRR03[np.newaxis, :]\n",
    "\n",
    "\n",
    "\n",
    "IWW_IRR_1901_2014_IPSL_IRR01 = IWW_IRR_1901_2014_IPSL_IRR01[np.newaxis, :]\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "IWW_IRR_1901_2014_CNRM_IRR01 = IWW_IRR_1901_2014_CNRM_IRR01[np.newaxis, :]\n",
    "\n",
    "IWW_NOI_1901_2014_CNRM_IRR01 = IWW_NOI_1901_2014_CNRM_NOI01[np.newaxis, :]\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e2a9d71b",
   "metadata": {},
   "outputs": [],
   "source": [
    "IWW_IRR_1901_2014_E3SM_IRR01 = IWW_IRR_1901_2014_E3SM_IRR01[np.newaxis, :]\n",
    "IWW_IRR_1901_2014_E3SM_IRR02 = IWW_IRR_1901_2014_E3SM_IRR02[np.newaxis, :]\n",
    "\n",
    "IWW_NOI_1901_2014_E3SM_IRR01 = IWW_IRR_1901_2014_E3SM_NOI01[np.newaxis, :]\n",
    "IWW_NOI_1901_2014_E3SM_IRR02 = IWW_IRR_1901_2014_E3SM_NOI02[np.newaxis, :]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e8217a93",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "\n",
    "IWW_IRR_1901_2014_ens = np.concatenate(((IWW_IRR_1901_2014_CESM2_IRR01[:,:-1]+\n",
    "                                        IWW_IRR_1901_2014_CESM2_IRR02[:,:-1]+\n",
    "                                        IWW_IRR_1901_2014_CESM2_IRR03[:,:-1])/3,\n",
    "                                        (IWW_IRR_1901_2014_CESM2_gw_IRR01[:,:-1]+IWW_IRR_1901_2014_CESM2_gw_IRR02[:,:-1]+IWW_IRR_1901_2014_CESM2_gw_IRR03[:,:-1])/3,\n",
    "                                        (IWW_IRR_1901_2014_NorESM_IRR01[:,:-1]+IWW_IRR_1901_2014_NorESM_IRR02[:,:-1]+IWW_IRR_1901_2014_NorESM_IRR03[:,:-1])/3,\n",
    "                                        IWW_IRR_1901_2014_IPSL_IRR01,\n",
    "                                        (IWW_IRR_1901_2014_E3SM_IRR01[:,:-1]+IWW_IRR_1901_2014_E3SM_IRR02[:,:-1])/2,\n",
    "                                        IWW_IRR_1901_2014_CNRM_IRR01),axis = 0)\n",
    "\n",
    "IWW_NOI_1901_2014_ens = np.concatenate(((IWW_NOI_1901_2014_CESM2_IRR01[:,:-1]+IWW_NOI_1901_2014_CESM2_IRR02[:,:-1]+IWW_NOI_1901_2014_CESM2_IRR03[:,:-1])/3,\n",
    "                                        (IWW_NOI_1901_2014_CESM2_gw_IRR01[:,:-1]+IWW_NOI_1901_2014_CESM2_gw_IRR02[:,:-1]+IWW_NOI_1901_2014_CESM2_gw_IRR03[:,:-1])/3,\n",
    "                                        (IWW_NOI_1901_2014_NorESM_IRR01[:,:-1]+IWW_NOI_1901_2014_NorESM_IRR03[:,:-1])/2,\n",
    "                                        (IWW_NOI_1901_2014_E3SM_IRR01[:,:-1]+IWW_NOI_1901_2014_E3SM_IRR02[:,:-1])/2,\n",
    "                                        IWW_NOI_1901_2014_CNRM_IRR01),axis = 0)\n",
    "\n",
    "IWW_IRR_1901_2014_25 = np.percentile(IWW_IRR_1901_2014_ens, 0, axis=0)\n",
    "IWW_IRR_1901_2014_50 = np.percentile(IWW_IRR_1901_2014_ens, 50, axis=0)\n",
    "IWW_IRR_1901_2014_75 = np.percentile(IWW_IRR_1901_2014_ens, 100, axis=0)\n",
    "\n",
    "IWW_NOI_1901_2014_25 = np.percentile(IWW_NOI_1901_2014_ens, 0, axis=0)\n",
    "IWW_NOI_1901_2014_50 = np.percentile(IWW_NOI_1901_2014_ens, 50, axis=0)\n",
    "IWW_NOI_1901_2014_75 = np.percentile(IWW_NOI_1901_2014_ens, 100, axis=0)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2a614467",
   "metadata": {},
   "outputs": [],
   "source": [
    "IWW_IRR_1901_2014_CESM2_WNA = calcu_global_IWW_region(str_IWW_CESM2, AREA, 4, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_CNA = calcu_global_IWW_region(str_IWW_CESM2, AREA, 5, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_NCA = calcu_global_IWW_region(str_IWW_CESM2, AREA, 7, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_MED = calcu_global_IWW_region(str_IWW_CESM2, AREA, 20, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_WCA = calcu_global_IWW_region(str_IWW_CESM2, AREA, 33, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_EAS = calcu_global_IWW_region(str_IWW_CESM2, AREA, 36, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_SAS = calcu_global_IWW_region(str_IWW_CESM2, AREA, 38, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_SEA = calcu_global_IWW_region(str_IWW_CESM2, AREA, 39, 'QIRRIG')\n",
    "\n",
    "IWW_IRR_1901_2014_CESM2_gw_WNA = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 4, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_gw_CNA = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 5, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_gw_NCA = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 7, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_gw_MED = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 20, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_gw_WCA = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 33, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_gw_EAS = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 36, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_gw_SAS = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 38, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_CESM2_gw_SEA = calcu_global_IWW_region(str_IWW_CESM2_gw, AREA, 39, 'QIRRIG')\n",
    "\n",
    "IWW_IRR_1901_2014_NorESM_WNA = calcu_global_IWW_region(str_IWW_NorESM, AREA, 4, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_NorESM_CNA = calcu_global_IWW_region(str_IWW_NorESM, AREA, 5, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_NorESM_NCA = calcu_global_IWW_region(str_IWW_NorESM, AREA, 7, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_NorESM_MED = calcu_global_IWW_region(str_IWW_NorESM, AREA, 20, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_NorESM_WCA = calcu_global_IWW_region(str_IWW_NorESM, AREA, 33, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_NorESM_EAS = calcu_global_IWW_region(str_IWW_NorESM, AREA, 36, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_NorESM_SAS = calcu_global_IWW_region(str_IWW_NorESM, AREA, 38, 'QIRRIG')\n",
    "IWW_IRR_1901_2014_NorESM_SEA = calcu_global_IWW_region(str_IWW_NorESM, AREA, 39, 'QIRRIG')\n",
    "\n",
    "IWW_IRR_1901_2014_E3SM_WNA = calcu_global_IWW_region(str_IWW_E3SM, AREA, 4, 'QIRRIG_REAL')\n",
    "IWW_IRR_1901_2014_E3SM_CNA = calcu_global_IWW_region(str_IWW_E3SM, AREA, 5, 'QIRRIG_REAL')\n",
    "IWW_IRR_1901_2014_E3SM_NCA = calcu_global_IWW_region(str_IWW_E3SM, AREA, 7, 'QIRRIG_REAL')\n",
    "IWW_IRR_1901_2014_E3SM_MED = calcu_global_IWW_region(str_IWW_E3SM, AREA, 20, 'QIRRIG_REAL')\n",
    "IWW_IRR_1901_2014_E3SM_WCA = calcu_global_IWW_region(str_IWW_E3SM, AREA, 33, 'QIRRIG_REAL')\n",
    "IWW_IRR_1901_2014_E3SM_EAS = calcu_global_IWW_region(str_IWW_E3SM, AREA, 36, 'QIRRIG_REAL')\n",
    "IWW_IRR_1901_2014_E3SM_SAS = calcu_global_IWW_region(str_IWW_E3SM, AREA, 38, 'QIRRIG_REAL')\n",
    "IWW_IRR_1901_2014_E3SM_SEA = calcu_global_IWW_region(str_IWW_E3SM, AREA, 39, 'QIRRIG_REAL')\n",
    "\n",
    "IWW_IRR_1901_2014_CNRM_WNA = calcu_global_IWW_region(str_IWW_CNRM, AREA, 4, 'airrww')\n",
    "IWW_IRR_1901_2014_CNRM_CNA = calcu_global_IWW_region(str_IWW_CNRM, AREA, 5, 'airrww')\n",
    "IWW_IRR_1901_2014_CNRM_NCA = calcu_global_IWW_region(str_IWW_CNRM, AREA, 7, 'airrww')\n",
    "IWW_IRR_1901_2014_CNRM_MED = calcu_global_IWW_region(str_IWW_CNRM, AREA, 20, 'airrww')\n",
    "IWW_IRR_1901_2014_CNRM_WCA = calcu_global_IWW_region(str_IWW_CNRM, AREA, 33, 'airrww')\n",
    "IWW_IRR_1901_2014_CNRM_EAS = calcu_global_IWW_region(str_IWW_CNRM, AREA, 36, 'airrww')\n",
    "IWW_IRR_1901_2014_CNRM_SAS = calcu_global_IWW_region(str_IWW_CNRM, AREA, 38, 'airrww')\n",
    "IWW_IRR_1901_2014_CNRM_SEA = calcu_global_IWW_region(str_IWW_CNRM, AREA, 39, 'airrww')\n",
    "\n",
    "IWW_IRR_1901_2014_IPSL_WNA = calcu_global_IWW_region(str_IWW_IPSL, AREA, 4, 'irr')\n",
    "IWW_IRR_1901_2014_IPSL_CNA = calcu_global_IWW_region(str_IWW_IPSL, AREA, 5, 'irr')\n",
    "IWW_IRR_1901_2014_IPSL_NCA = calcu_global_IWW_region(str_IWW_IPSL, AREA, 7, 'irr')\n",
    "IWW_IRR_1901_2014_IPSL_MED = calcu_global_IWW_region(str_IWW_IPSL, AREA, 20, 'irr')\n",
    "IWW_IRR_1901_2014_IPSL_WCA = calcu_global_IWW_region(str_IWW_IPSL, AREA, 33, 'irr')\n",
    "IWW_IRR_1901_2014_IPSL_EAS = calcu_global_IWW_region(str_IWW_IPSL, AREA, 36, 'irr')\n",
    "IWW_IRR_1901_2014_IPSL_SAS = calcu_global_IWW_region(str_IWW_IPSL, AREA, 38, 'irr')\n",
    "IWW_IRR_1901_2014_IPSL_SEA = calcu_global_IWW_region(str_IWW_IPSL, AREA, 39, 'irr')\n",
    "\n",
    "# IWW_IRR_1901_2014_MIROC_WNA = calcu_global_IWW_region(str_IWW_MIROC, AREA, 4, 'irrac')\n",
    "# IWW_IRR_1901_2014_MIROC_CNA = calcu_global_IWW_region(str_IWW_MIROC, AREA, 5, 'irrac')\n",
    "# IWW_IRR_1901_2014_MIROC_NCA = calcu_global_IWW_region(str_IWW_MIROC, AREA, 7, 'irrac')\n",
    "# IWW_IRR_1901_2014_MIROC_MED = calcu_global_IWW_region(str_IWW_MIROC, AREA, 20, 'irrac')\n",
    "# IWW_IRR_1901_2014_MIROC_WCA = calcu_global_IWW_region(str_IWW_MIROC, AREA, 33, 'irrac')\n",
    "# IWW_IRR_1901_2014_MIROC_EAS = calcu_global_IWW_region(str_IWW_MIROC, AREA, 36, 'irrac')\n",
    "# IWW_IRR_1901_2014_MIROC_SAS = calcu_global_IWW_region(str_IWW_MIROC, AREA, 38, 'irrac')\n",
    "# IWW_IRR_1901_2014_MIROC_SEA = calcu_global_IWW_region(str_IWW_MIROC, AREA, 39, 'irrac')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "284201ba",
   "metadata": {},
   "outputs": [],
   "source": [
    "def plot_fill_between(ax, index, data1, data2, title, title1, label, color, ylabel, xlabel, marker):\n",
    "    ax.text(-0.05, 1.05, index, color='dimgrey', fontsize=12, transform=ax.transAxes, weight='bold')\n",
    "    # ax.step(range(1901, 2015), data, label = label, color = color, linewidth = 1, marker = marker, markersize=1.2)\n",
    "    ax.fill_between(range(1901, 2015), data1, data2, label = label, color = color, linewidth = 1, alpha=0.8)\n",
    "    plt.title(title, loc='right')\n",
    "    plt.title(title1, loc='left')\n",
    "    plt.ylabel(ylabel, fontsize = 14)\n",
    "    plt.xlabel(xlabel, fontsize = 14)\n",
    "    plt.yticks(fontsize = 12)\n",
    "    plt.xticks(fontsize=12)\n",
    "    \n",
    "    \n",
    "def plot_fill_between_all(ax, index, data1, data2, data3, data4, data5, data6, data7, data8, data9, ylabel, ylim_max):\n",
    "    plot_fill_between(ax, '', data1 + data2 + data3 + data4 + data5 + data6 + data7 + data8, data9, '', '', 'Other', 'chocolate', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, '', data1 + data2 + data3 + data4 + data5 + data6 + data7, data1 + data2 + data3 + data4 + data5 + data6 + data7 + data8, '', '', 'WNA', 'b', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, '', data1 + data2 + data3 + data4 + data5 + data6, data1 + data2 + data3 + data4 + data5 + data6 + data7, '', '', 'CNA', 'c', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, '', data1 + data2 + data3 + data4 + data5, data1 + data2 + data3 + data4 + data5 + data6, '', '', 'NCA', 'g', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, '', data1 + data2 + data3 + data4, data1 + data2 + data3 + data4 + data5, '', '', 'MED', 'm', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, '', data1 + data2 + data3, data1 + data2 + data3 + data4, '', '', 'WCA', 'r', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, '', data1 + data2, data1 + data2 + data3, '', '', 'EAS', 'y', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, '', data1, data1 + data2, '', '', 'SAS', 'orange', ylabel, 'year', '^')\n",
    "    plot_fill_between(ax, index, 0, data1, '', '', 'SEA', 'pink', ylabel, 'year', '^')\n",
    "    ax1.set_ylim(0,ylim_max)\n",
    "    ax1.set_xlim(1901,2014)\n",
    "    \n",
    "def plot_line(ax, index, data, title, title1, label, color, ylabel, xlabel, marker):\n",
    "    ax.text(-0.05, 1.05, index, color='dimgrey', fontsize=12, transform=ax.transAxes, weight='bold')\n",
    "    # ax.step(range(1901, 2015), data, label = label, color = color, linewidth = 1, marker = marker, markersize=1.2)\n",
    "    ax.plot(range(1901, 2015), data, label = label, color = color, linewidth = 2, markersize=1.2, alpha=0.8)\n",
    "    plt.title(title, loc='right')\n",
    "    plt.title(title1, loc='left')\n",
    "    plt.ylabel(ylabel, fontsize = 14)\n",
    "    plt.xlabel(xlabel, fontsize = 14)\n",
    "    plt.yticks(fontsize = 12)\n",
    "    plt.xticks(fontsize=12)\n",
    "    plt.xlim(1901,2015)\n",
    "    plt.legend(loc='upper left')\n",
    "\n",
    "fig = plt.figure(figsize=(20, 8),dpi=300)\n",
    "fig.subplots_adjust(hspace=0.3, wspace=0.3, left = 0.05, right = 0.95, top = 0.95, bottom = 0.05)\n",
    "\n",
    "ax1 = plt.subplot(241, frameon=True)\n",
    "ax1.spines[\"top\"].set_visible(True)\n",
    "ax1.spines[\"right\"].set_visible(True)\n",
    "plot_fill_between_all(ax1, 'a', global_irri_land_SEA[:] / 1000000, global_irri_land_SAS[:] / 1000000, global_irri_land_EAS[:] / 1000000, global_irri_land_WCA[:] / 1000000, global_irri_land_MED[:] / 1000000, global_irri_land_NCA[:] / 1000000, global_irri_land_CNA[:] / 1000000, global_irri_land_WNA[:] / 1000000, global_irri_land[:] / 1000000, 'AEI ($\\mathregular{10^6}$ $\\mathregular{km^2}$)', 3)    \n",
    "plt.grid(linestyle = '--')\n",
    "plt.legend(loc = 'upper left')\n",
    "\n",
    "# ax1 = plt.subplot2grid((2, 3), (0, 1), colspan = 2)\n",
    "ax1 = plt.subplot(242, frameon=True)\n",
    "ax1.text(-0.05, 1.05, 'b', color='dimgrey', fontsize=12, transform=ax1.transAxes, weight='bold')\n",
    "\n",
    "plot_line(ax1, '', IWW_IRR_1901_2014_50, '', '', 'tranirr', 'b', '', 'year', '^')\n",
    "plot_line(ax1, '', IWW_NOI_1901_2014_50, '', '', '1901irr', 'r', '', 'year', '>')\n",
    "plt.ylabel('IWW ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1.fill_between(range(1901, 2015), IWW_IRR_1901_2014_25, IWW_IRR_1901_2014_75, color = 'b', linewidth = 2, alpha=0.2)\n",
    "ax1.fill_between(range(1901, 2015), IWW_NOI_1901_2014_25, IWW_NOI_1901_2014_75, color = 'r', linewidth = 2, alpha=0.2)\n",
    "ax1.set_ylim(0,4500)\n",
    "plt.grid(linestyle = '--')\n",
    "\n",
    "ax1 = plt.subplot(243, frameon=True)\n",
    "# plt.ylabel('Irrigation water($\\mathregular{km^3}$)', fontsize = 14)\n",
    "plot_fill_between_all(ax1, 'c CESM2', IWW_IRR_1901_2014_CESM2_SEA[:-1], IWW_IRR_1901_2014_CESM2_SAS[:-1], IWW_IRR_1901_2014_CESM2_EAS[:-1], IWW_IRR_1901_2014_CESM2_WCA[:-1], IWW_IRR_1901_2014_CESM2_MED[:-1], IWW_IRR_1901_2014_CESM2_NCA[:-1], IWW_IRR_1901_2014_CESM2_CNA[:-1], IWW_IRR_1901_2014_CESM2_WNA[:-1], IWW_IRR_1901_2014_CESM2[:-1], '', 1100)    \n",
    "plt.grid(linestyle = '--')\n",
    "plt.ylabel('IWW ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1.set_ylim(0,4500)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1 = plt.subplot(244, frameon=True)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "plot_fill_between_all(ax1, 'd CESM2_gw', IWW_IRR_1901_2014_CESM2_gw_SEA[:-1], IWW_IRR_1901_2014_CESM2_gw_SAS[:-1], IWW_IRR_1901_2014_CESM2_gw_EAS[:-1], IWW_IRR_1901_2014_CESM2_gw_WCA[:-1], IWW_IRR_1901_2014_CESM2_gw_MED[:-1], IWW_IRR_1901_2014_CESM2_gw_NCA[:-1], IWW_IRR_1901_2014_CESM2_gw_CNA[:-1], IWW_IRR_1901_2014_CESM2_gw_WNA[:-1], IWW_IRR_1901_2014_CESM2_gw[:-1], '', 1100)    \n",
    "plt.grid(linestyle = '--')\n",
    "plt.ylabel('IWW ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1.set_ylim(0,4500)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1 = plt.subplot(245, frameon=True)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "plot_fill_between_all(ax1, 'e NorESM', IWW_IRR_1901_2014_NorESM_SEA[:-1], IWW_IRR_1901_2014_NorESM_SAS[:-1], IWW_IRR_1901_2014_NorESM_EAS[:-1], IWW_IRR_1901_2014_NorESM_WCA[:-1], IWW_IRR_1901_2014_NorESM_MED[:-1], IWW_IRR_1901_2014_NorESM_NCA[:-1], IWW_IRR_1901_2014_NorESM_CNA[:-1], IWW_IRR_1901_2014_NorESM_WNA[:-1], IWW_IRR_1901_2014_NorESM[:-1], '', 1100)    \n",
    "plt.grid(linestyle = '--')\n",
    "plt.ylabel('IWW ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1.set_ylim(0,4500)\n",
    "ax1 = plt.subplot(246, frameon=True)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "plot_fill_between_all(ax1, 'f E3SM', IWW_IRR_1901_2014_E3SM_SEA[:-1], IWW_IRR_1901_2014_E3SM_SAS[:-1], IWW_IRR_1901_2014_E3SM_EAS[:-1], IWW_IRR_1901_2014_E3SM_WCA[:-1], IWW_IRR_1901_2014_E3SM_MED[:-1], IWW_IRR_1901_2014_E3SM_NCA[:-1], IWW_IRR_1901_2014_E3SM_CNA[:-1], IWW_IRR_1901_2014_E3SM_WNA[:-1], IWW_IRR_1901_2014_E3SM[:-1], '', 1100)    \n",
    "plt.grid(linestyle = '--')\n",
    "plt.ylabel('IWW ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1.set_ylim(0,4500)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1 = plt.subplot(247, frameon=True)\n",
    "\n",
    "# plot_fill_between_all(ax1, 'c MIROC', IWW_IRR_1901_2014_MIROC_SEA, IWW_IRR_1901_2014_MIROC_SAS, IWW_IRR_1901_2014_MIROC_EAS, IWW_IRR_1901_2014_MIROC_WCA, IWW_IRR_1901_2014_MIROC_MED, IWW_IRR_1901_2014_MIROC_NCA, IWW_IRR_1901_2014_MIROC_CNA, IWW_IRR_1901_2014_MIROC_WNA, IWW_IRR_1901_2014_MIROC, 'Irrigation water($\\mathregular{km^3}$)', 1100)    \n",
    "# plt.grid(linestyle = '--')\n",
    "# ax1.set_ylim(0,2000)\n",
    "# ax1 = plt.subplot(338, frameon=True)\n",
    "\n",
    "plot_fill_between_all(ax1, 'g CNRM-CM6', IWW_IRR_1901_2014_CNRM_SEA, IWW_IRR_1901_2014_CNRM_SAS, IWW_IRR_1901_2014_CNRM_EAS, IWW_IRR_1901_2014_CNRM_WCA, IWW_IRR_1901_2014_CNRM_MED, IWW_IRR_1901_2014_CNRM_NCA, IWW_IRR_1901_2014_CNRM_CNA, IWW_IRR_1901_2014_CNRM_WNA, IWW_IRR_1901_2014_CNRM, '', 1100)    \n",
    "plt.grid(linestyle = '--')\n",
    "plt.ylabel('IWW ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1.set_ylim(0,4500)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1 = plt.subplot(248, frameon=True)\n",
    "\n",
    "plot_fill_between_all(ax1, 'h IPSL-CM6', IWW_IRR_1901_2014_IPSL_SEA, IWW_IRR_1901_2014_IPSL_SAS, IWW_IRR_1901_2014_IPSL_EAS, IWW_IRR_1901_2014_IPSL_WCA, IWW_IRR_1901_2014_IPSL_MED, IWW_IRR_1901_2014_IPSL_NCA, IWW_IRR_1901_2014_IPSL_CNA, IWW_IRR_1901_2014_IPSL_WNA, IWW_IRR_1901_2014_IPSL, '', 1100)    \n",
    "plt.grid(linestyle = '--')\n",
    "plt.ylabel('IWW ($\\mathregular{km^3/yr}$)', fontsize = 14)\n",
    "ax1.set_ylim(0,4500)\n",
    "# plt.ylabel('Irrigation water \\n withdrawal ($\\mathregular{km^3/yr}$)', fontsize = 14)"
   ]
  }
 ],
 "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.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}