A one-hot-encoder for DNA sequences that I find myself writing repeatedly

one_hot_encode.py

def one_hot_encode(sequence: str, alphabet=["G", "A", "T", "C"]):
    """
    one-hot encode a string using a pre-defined alphabet

    >>> one_hot_encode("GATC")
    [1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1]
    """

    vector = ([0] * len(alphabet)) * len(sequence)

    for n, character in enumerate(sequence):
        vector[alphabet.index(character) + (n * len(alphabet))] = 1

    return vector