版权所有 ＱＱ：２３１４６９２４２
#coding=utf-8
#孟德尔自由组合定理
import copy
#大写字母表示显性基因，小写字母表示隐性基因
#gene,如D或d
#基因型genotype
genotype11=['Y','Y']
genotype12=['y','y']
genotype13=['Y','y']
genotype21=['R','R']
genotype22=['r','r']
genotype23=['R','r']
DNA1=[genotype13,genotype23]
DNA2=[genotype12,genotype22]
DNA3=[genotype11,genotype21]
DNA4=[genotype12,genotype22]
DNA_group=[DNA3,DNA4]
#基因表现型character
#字母大小写判断a.islower()，a.isupper()
def Character(genotype1):
for i in genotype1:
if i.isupper()==True:
return "Dominant"
return "recessive"
#Phenotype输入基因型，返回它的代表基因
def Phenotype(genotype1):
for i in genotype1:
if i.isupper()==True:
return i
return genotype1[0]
#配子后，基因型genotype的可能数量
def Assort(genotype1,genotype2):
genotype_list=[]
for gene1 in genotype1:
for gene2 in genotype2:
genotype=[gene1,gene2]
genotype_list.append(genotype)
return genotype_list
#返回不重复列表
def None_repeat_list(list1):
none_repeat_list=[]
for i in list1:
#print 'i:',i
if i not in none_repeat_list:
none_repeat_list.append(i)
return none_repeat_list
#计算各个基因类型比例
def Count_list(none_repeat_list,DNA_sort_list,n):
count_list=[]
for i in none_repeat_list:
count=DNA_sort_list.count(i)
count=count*1.0/n
count_list.append(count)
return count_list
#返回各个基因类型比例
def Ratio_genotype(none_repeat_list,count_list):
ratio_genotype=zip(none_repeat_list,count_list)
return ratio_genotype
#多个基因型的自由组合
def Independent_assort(DNA1,DNA2):
genotype00=DNA1[0]
genotype01=DNA2[0]
assortment1=Assort(genotype00,genotype01)
genotype10=DNA1[1]
genotype11=DNA2[1]
assortment2=Assort(genotype10,genotype11)
result=Assort(assortment1,assortment2)
return result
#排序函数,大写字母（线性基因）在前面.例如[r,R]排序后变成[R,r]
def Sort(list1):
k=0
n=len(list1[0])
#print "n:",n
for i in range(len(list1)):
#print 'i',i
while k<n:
#print 'k',k
#print 'list1[i][k]',list1[i][k]
list1[i][k].sort()
k+=1
k=0
return list1
#DNA所有性状结果
'''>>> Character_DNA(DNA_noneRepeat_assortment)
[['Y', 'R'], ['Y', 'R'], ['Y', 'r'], ['Y', 'R'], ['Y', 'R'], ['Y', 'r'], ['y', 'R'], ['y', 'R'], ['y', 'r']]'''
def Character_DNA(DNA_noneRepeat_assortment):
character_DNA=copy.deepcopy(DNA_noneRepeat_assortment)
k=0
noneRepeat_num=len(DNA_noneRepeat_assortment)
element_num=len(DNA_noneRepeat_assortment[0])
#print "n:",n
for i in range(noneRepeat_num):
#print 'i',i
while k<element_num:
#print 'k',k
#print 'list1[i][k]',list1[i][k]
character_DNA[i][k]=Phenotype(character_DNA[i][k])
k+=1
k=0
return character_DNA
#最终性状和对应数量比
def NoneRepeat_character_ratio(ratio_character,noneRepeat_character_DNA):
num=len(ratio_character)
list1=copy.deepcopy(ratio_character)
list_num=[]
count=0
for i in noneRepeat_character_DNA:
for k in list1:
if k[0]==i:
count+=k[1]
#print "count",count
list_num.append(count)
count=0
return list_num
def Analysis(DNA_group,ratio_character_result):
print "DNA_group:",DNA_group
print 'ratio_character_result:',ratio_character_result
#多个基因型的自由组合结果（有重复）
DNA_assortment=Independent_assort(DNA3,DNA4)
#多基因自由组合后的排序，大写字母（线性基因）在前面.例如[r,R]排序后变成[R,r]
DNA_sort=Sort(DNA_assortment)
#自由组合个数
n=len(DNA_sort)
#多个基因型的自由组合结果（无重复）
DNA_noneRepeat_assortment=None_repeat_list(DNA_sort)
#排序结果数量统计
count_list=Count_list(DNA_noneRepeat_assortment,DNA_sort,n)
#排序后基因类型对应的数量统计
ratio_genotype=Ratio_genotype(DNA_noneRepeat_assortment,count_list)
#DNA性状结果(重复)
character_DNA=Character_DNA(DNA_noneRepeat_assortment)
#数量统计
ratio_character=Ratio_genotype(character_DNA,count_list)
#DNA性状结果（去重复）
noneRepeat_character_DNA=None_repeat_list(character_DNA)
#DNA性状和数量比（不重复）
count_noneRepeat_character=NoneRepeat_character_ratio(ratio_character,noneRepeat_character_DNA)
#DNA性状和数量比,最后整合
ratio_character_result=Ratio_genotype(noneRepeat_character_DNA,count_noneRepeat_character)
#分析结果
Analysis(DNA_group,ratio_character_result)