import sys, datetime, copy, os
from dateutil.relativedelta import relativedelta
import numpy as np; np.set_printoptions(threshold=sys.maxsize)
import matplotlib; matplotlib.use('Agg')
import matplotlib.pyplot as plt
import matplotlib.colors as clr
from mpl_toolkits.basemap import Basemap

import seaborn as sns
from matplotlib.colors import ListedColormap
from netCDF4 import Dataset

nc   = Dataset('NC/percentile_sw_1961-2020_daily_Reg.nc','r')
day  = nc.variables['day'][:]
reg  = nc.variables['reg'][:]
sw1  = nc.variables['sw1'][:]

xlabels = ['R1','R2','R3','R4','R5','R6','R7']

fig, ax = plt.subplots(7,figsize=(6,10))
plt.subplots_adjust(hspace=0.5)

for k in range(len(reg)):
    x = day[0:30]
    y = sw1[0:30,k]
    fit = np.polyfit(x, y, 1)

    ax[k].scatter(x, y, s=6)
    ax[k].set_ylim(0,100)
    ax[k].plot(x, fit[0]*x + fit[1], 'r-')
    ax[k].text(22, 90, 'y='+str(round(fit[0],2))+'*x + '+str(round(fit[1],2)))

fig.savefig('scatter_demo.pdf', bbox_inches="tight", pad_inches=.05, dpi=600)