INNER CODE UNIT · Python
cluster_allele_counts_soup
wheaton5/souporcell · consensus.py:297
cluster_allele_counts_soup = [[[0,0] for c in range(max_cluster+1)] for x in range(len(loci_for_soup))]
average_allele_expression_soup = [[0,0] for c in range(len(loci_for_soup))]
average_allele_expression = [[0,0] for c in range(total_loci)]
for (cell, loci_counts) in cell_counts.items():
if cell-1 in doublets:
continue
cluster = cell_clusters[cell-1]
for (locus, counts) in loci_counts.items():
ref = 0
alt = 0
if counts[0] > 0:
ref = counts[0]
if counts[1] > 0:
alt = counts[1]
if locus-1 in loci_for_soup:
cluster_allele_counts_soup[loci_for_soup[locus-1]][cluster][0] += ref
cluster_allele_counts_soup[loci_for_soup[locus-1]][cluster][1] += alt
average_allele_expression_soup[loci_for_soup[locus-1]][0] += ref