2022-09-14 ROSLIND_7: Mendel's First Law

# ----ROSLIND_7: ---- 
# Mendel's First Law
"""
重点：
1. 不放回抽样
2. Aa X aa 时注意是两次,前后位置交换后是另外一次
3. 种群中某一种基因型的概率 * 另一种基因型的概率 * 遗传概率
"""

def comp_prop(k, m, n):
    t = k + m + n # 定义总体人数
    # 算出隐形基因型的概率 1 - aa*aa - Aa*Aa - Aa*aa - aa*Aa
    prop = 1 - n/t * (n-1)/(t-1) - m/t * (m-1)/(t-1) * 0.25 - 2 * n/t * m/(t-1) * 0.5
    return round(prop,5)