利用朴素贝叶斯算法解决“公园凉鞋问题”

问题描述

在夏季，某公园男性穿凉鞋的概率为 12 ，女性穿凉鞋的概率为 23 ，并且该公园中男女比例通常为 2:1 ，问题：若你在公园中随机遇到一个穿凉鞋的人，请问他的性别为男性或女性的概率分别为多少？

计算过程

代码实现

P_men, P_women = 2 / 3, 1 / 3  # 男女比例
P_men_true, P_women_true = 1 / 2, 2 / 3  # 男女穿凉鞋的概率
P_true = P_men * P_men_true + P_women * P_women_true  # 整个公园男女穿凉鞋的概率


def sandal_naive_bayes(p_true, p_men, p_women, p_men_true, p_women_true):
    p1 = (p_men_true * p_men) / p_true
    p2 = (p_women_true * p_women) / p_true
    print(f'在已知此人穿凉鞋的情况下，此人为男性的概率是：{p1}， 为女性的概率是：{p2}')


if __name__ == '__main__':
    sandal_naive_bayes(P_true, P_men, P_women, P_men_true, P_women_true)

运行结果