diff --git a/tupak/core/result.py b/tupak/core/result.py index 20f7e1ba0029fba8677b49ad790899522786aa9f..194ed50f707dfbf5d3c567c1ee7c2ae2cd1b0d06 100644 --- a/tupak/core/result.py +++ b/tupak/core/result.py @@ -10,6 +10,7 @@ from collections import OrderedDict from tupak.core import utils from tupak.core.utils import logger +from tupak.core.prior import DeltaFunction def result_file_name(outdir, label): @@ -435,6 +436,24 @@ class Result(dict): data_frame = conversion_function(data_frame, likelihood, priors) self.posterior = data_frame + def calculate_prior_values(self, priors): + """ + Evaluate prior probability for each parameter for each sample. + + Parameters + ---------- + priors: dict, PriorSet + Prior distributions + """ + self.prior_values = pd.DataFrame() + for key in priors: + if key in self.posterior.keys(): + if isinstance(priors[key], DeltaFunction): + continue + else: + self.prior_values[key]\ + = priors[key].prob(self.posterior[key].values) + def construct_cbc_derived_parameters(self): """ Construct widely used derived parameters of CBCs """ self.posterior['mass_chirp'] = (