Bayes' rule | Python