diff --git a/vcf_to_sfs.py b/vcf_to_sfs.py index 64614d8..fcc49c0 100755 --- a/vcf_to_sfs.py +++ b/vcf_to_sfs.py @@ -82,16 +82,16 @@ with gzip.open(sys.argv[1], "rb") as inputgz: if folded : #allele_counts_list = list(allele_counts.keys()) ##print("ALC", allele_counts_list, "POLY", polyallelic, ALT) - # for al in range(polyallelic-1): - # SFS_values[min(allele_counts_list)-1] += 1/len(ALT) - # allele_counts_list.remove(min(allele_counts_list)) - if len(ALT) == 1: - SFS_values[min(allele_counts_list)-1] += 1 - else: - for al in range(polyallelic-1): - SFS_values[min(allele_counts_list)-1] += 1/len(ALT) - allele_counts_list.remove(min(allele_counts_list)) - polycount += 1 + for al in range(polyallelic-1): + SFS_values[min(allele_counts_list)-1] += 1/len(ALT) + allele_counts_list.remove(min(allele_counts_list)) + # if len(ALT) == 1: + # SFS_values[min(allele_counts_list)-1] += 1 + # else: + # for al in range(polyallelic-1): + # SFS_values[min(allele_counts_list)-1] += 1/len(ALT) + # allele_counts_list.remove(min(allele_counts_list)) + # polycount += 1 line = inputgz.readline() print(SFS_values) print(polycount)