Ребята в детском саду учатся считать, и чтобы им было интереснее, воспитательница решила оформить список изучаемых чисел

n = int(input("Введите высоту прямоугольника: "))
m = int(input("Введите ширину прямоугольника: "))
 
# Создаем пустой двумерный массив размером n x m
snake = [[0] * m for _ in range(n)]
 
# Задаем начальное значение и направление движения
num = 1
direction = 1 # 1 - вправо, -1 - влево
 
# Обходим прямоугольник змейкой и заполняем его числами
for i in range(n):
    # Если направление движения вправо, то заполняем строки слева направо
    if direction == 1:
        for j in range(m):
            snake[i][j] = num
            num += 1
        direction = -1 # После заполнения строки меняем направление на влево
    # Если направление движения влево, то заполняем строки справа налево
    else:
        for j in range(m - 1, -1, -1):
            snake[i][j] = num
            num += 1
        direction = 1 # После заполнения строки меняем направление на вправо
 
# Выводим змейку на экран, выравнивая столбцы по ширине самого длинного числа
max_num_width = len(str(n * m)) # Длина самого длинного числа
for i in range(n):
    for j in range(m):
        print(str(snake[i][j]).rjust(max_num_width), end=" ")
    print() # Переходим на новую строку после каждой строки змейки

n = int(input("Введите высоту прямоугольника: "))
m = int(input("Введите ширину прямоугольника: "))
l = [[i * m + j + 1 for j in range(m)] for i in range(n)]
for i in range(1, n, 2):
    l[i] = l[i][::-1]
for i in range(n):
    for j in range(m):
        print(f'{l[i][j]:4d}', end='')
    print()