def map_alleles(self, mapping, **kwargs):
def f(block, bmapping):
return block.map_alleles(bmapping, copy=False)
domain = (self, mapping)
out = _chunked.map_blocks(domain, f, **kwargs)
return HaplotypeChunkedArray(out)
def map_alleles(self, mapping, **kwargs):
def f(block, bmapping):
return block.map_alleles(bmapping)
domain = (self, mapping)
out = _chunked.map_blocks(domain, f, **kwargs)
return AlleleCountsChunkedArray(out)
def count_alleles_subpops(self, subpops, max_allele=None,
**kwargs):
if max_allele is None:
max_allele = self.max()
def f(block):
return block.count_alleles_subpops(subpops, max_allele=max_allele)
out = _chunked.map_blocks(self, f, create='table', **kwargs)
return AlleleCountsChunkedTable(out)
def map_alleles(self, mapping, max_allele=None, **kwargs):
if max_allele is None:
max_allele = np.max(mapping)
def f(block, bmapping):
return block.map_alleles(bmapping, max_allele=max_allele)
domain = (self, mapping)
out = _chunked.map_blocks(domain, f, **kwargs)
return AlleleCountsChunkedArray(out)
def from_packed(cls, packed, **kwargs):
def f(block):
return GenotypeArray.from_packed(block)
out = _chunked.map_blocks(packed, f, **kwargs)
return cls(out)
