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]')