from typing import Sequence


def map_context(index: int, context_count: int) -> Sequence[float]:
    """Get a one-hot encoding for the given context."""
    context_vector = [0.0 for _ in range(context_count)]
    context_vector[index] = 1.0
    return context_vector