Write a function to generate a Latin Square given a positive integer `n`. The values can be any n distinct values, and don't have to be consistent for different n.

latin_square.py

def latin_square(n):
    if n <= 0:
        return []

    numbers = list(range(1, n + 1))
    result = []

    for i in range(n):
        result.append(numbers.copy())
        # Rotate left: move first element to the end
        numbers = numbers[1:] + [numbers[0]]

    return result


if __name__ == "__main__":
    print(latin_square(1))
    print(latin_square(2))
    print(latin_square(3))
    print(latin_square(4))

output.txt

$uv run python latin_square.py
[[1]]
[[1, 2], [2, 1]]
[[1, 2, 3], [2, 3, 1], [3, 1, 2]]
[[1, 2, 3, 4], [2, 3, 4, 1], [3, 4, 1, 2], [4, 1, 2, 3]]