def list_probs(wp_sm_assumption, func, func_args):
allowed_wp_sm = fwd_model.wp_sm_combo_options(wp_sm_assumption)
for wp_sm in allowed_wp_sm:
prob = func(wp_sm[0], wp_sm[1], **func_args)
if prob > 0:
print(wp_sm, prob)
return wp_sm
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))
