Exemple #1
0
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))
Exemple #2
0
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)