def load_and_plot_diffs_more_cases(varl, cases, case_oth, startyear, endyear,
pressure_adjust=pressure_adjust,
relative=False,
cbar_orientation='vertical',
asp_ratio=2, subfig_size=3,
ylim=None,
yscale='log',
cbar_eq=True,
norm=None):
if ylim is None:
ylim = [1e3, 100]
imp_cases = list(set(cases).union({case_oth}))
print(imp_cases)
cases_dic = get_averaged_fields.get_levlat_cases(imp_cases, varl, startyear, endyear,
pressure_adjust=pressure_adjust)
ncol = len(cases)
nrow = len(varl)
figsize = [subfig_size * ncol * asp_ratio, subfig_size * nrow]
# noinspection PyTypeChecker
fig, axs = plt.subplots(nrow, ncol, figsize=figsize, sharex=True, sharey=True)
for j, var in enumerate(varl):
saxs = axs[j, :]
levlat_more_cases_var(var, case_oth, cases, cases_dic, cbar_eq, cbar_orientation, saxs, norm, relative, ylim,
yscale)
return axs
def load_and_plot_diffs(varl, case_ctrl, case_other, start_time, end_time,
pressure_coords=True,
relative=False,
cbar_orient='vertical',
asp_ratio=2, subfig_size=3,
ncol=None,
ylim=None,
yscale='log',
norm=None
):
if ylim is None:
ylim = [1e3, 100]
cases_dict = get_averaged_fields.get_levlat_cases(cases, varl, start_time, end_time,
pressure_adjust=pressure_coords)
_nv = len(varl)
if ncol is None:
if _nv > 3:
ncol = 2
else:
ncol = 1
# noinspection PyUnresolvedReferences
nrow = int(np.ceil(_nv / ncol))
figsize = [subfig_size * ncol * asp_ratio, subfig_size * nrow]
# noinspection PyTypeChecker
fig, axs = plt.subplots(nrow, ncol, figsize=figsize, sharex=True, sharey=True)
for ax, var in zip(axs.flat, varl):
plot_levlat_diff(var, case_ctrl, case_other, cases_dict,
cbar_orientation=cbar_orient,
relative=relative,
ylim=ylim,
yscale=yscale,
ax=ax,
norm=norm)
return axs
case_oth = cases_sec[0]
ncol = len(cases_ctrl)
nrow = len(varlist)
subfig_size = 2.6
asp_ratio = 1.6
figsize = [subfig_size * ncol * asp_ratio, subfig_size * nrow]
# noinspection PyTypeChecker
fig, axs = plt.subplots(nrow, ncol, figsize=figsize, sharex=True, sharey=True)
norm_dic = dict(
NCONC01=colors.SymLogNorm(vmin=-1e3, vmax=1e3, linthresh=10),
NMR01=colors.SymLogNorm(vmin=-10, vmax=10, linthresh=1),# linscale=.5),
AWNC_incld=colors.SymLogNorm(vmin=-20, vmax=20, linthresh=1),
AREL_incld=colors.SymLogNorm(vmin=-5, vmax=5, linthresh=.1)
)
cases_dic = get_averaged_fields.get_levlat_cases(cases, varlist, startyear, endyear,
pressure_adjust=pressure_adjust)
for j, var in enumerate(varlist):
saxs = axs[j, :]
levlat_more_cases_var(var, case_oth, cases_ctrl, cases_dic, cbar_eq=True,
cbar_orientation='vertical',
axs=saxs,
norm=norm_dic[var],
relative=False,
ylim=[1e3, 200],
yscale='log')
for ax in axs.flatten():
ax.set_ylabel('')
ax.set_xlabel('')
for ax in axs[:, 0]:
ax.set_ylabel('Pressure [hPa]')