def overall_plot_wp_sm(title, func, func_args, wp_sm_assumption): plt.figure("hello") possible_wp_lst = fwd_model.make_wp_lst() possible_sm_lst = fwd_model.make_sm_lst() wp_sm = fwd_model.wp_sm_combo_options(wp_sm_assumption) plt.subplots_adjust(bottom=0.05, top=.95) to_plot = [0] * len(possible_wp_lst) for cnt, pwp in enumerate(possible_wp_lst): to_plot[cnt] = [func(pwp, psm, **func_args) if (pwp, psm) in wp_sm else 0 for psm in possible_sm_lst] flattened = [elt for sublist in to_plot for elt in sublist] overall_min = min(flattened) overall_max = max(flattened) for cnt, pwp in enumerate(possible_wp_lst): plt.subplot(3,3,cnt+1) plotting_wp(to_plot[cnt], pwp[0], (overall_min, overall_max))
def overall_plot_sm(func, func_args): possible_sm_lst = fwd_model.make_sm_lst() to_plot = [func(psm, func_args) for psm in possible_sm_lst] plotting_sm(to_plot)