我有一份清单清单:
sample = [['TTTT', 'CCCZ'], ['ATTA', 'CZZC']]
count = [[4,3],[4,2]]
correctionfactor = [[1.33, 1.5],[1.33,2]]
我计算每个字符(pi)的频率,将其平方然后求和(然后我计算het = 1 – sum).
The desired output [[1,2],[1,2]] #NOTE: This is NOT the real values of expected output. I just need the real values to be in this format.
问题:我不知道如何在这个循环中传递列表列表(sample,count)来提取所需的值.我以前只使用此代码传递了一个列表(例如[‘TACT’,’TTTT’..]).
>我怀疑我需要添加一个更大的for循环,对样本中的每个元素进行索引(即样本上的索引[0] = [‘TTTT’,’CCCZ’]和样本[1] = [‘ATTA’,’ CZZC’].我不知道如何将其纳入代码中.
**
码
list_of_hets = []
for idx, element in enumerate(sample):
count_dict = {}
square_dict = {}
for base in list(element):
if base in count_dict:
count_dict[base] += 1
else:
count_dict[base] = 1
for allele in count_dict: #Calculate frequency of every character
square_freq = (count_dict[allele] / count[idx])**2 #Square the frequencies
square_dict[allele] = square_freq
pf = 0.0
for i in square_dict:
pf += square_dict[i] # pf --> pi^2 + pj^2...pn^2 #Sum the frequencies
het = 1-pf
list_of_hets.append(het)
print list_of_hets
"Failed" OUTPUT:
line 70, in
square_freq = (count_dict[allele] / count[idx])**2
TypeError: unsupported operand type(s) for /: 'int' and 'list'er