dna manipulation functions

dna_functions.py

from functional import seq # PyFunctional
from itertools import product

def all_kmers(k):
  '''return list of all dna carthesien products of length k'''
  all = list(product('ACGT', repeat=k))
  return seq(all).map(lambda x: ''.join(x)).to_list()
  
def kmer_per_segment(dna_segment, k):
  '''return all the dna substrings of length k of the different dna strings in dna list'''
  return [dna_segment[i:i+k] for i in range(len(dna_segment)-(k-1))]