Заполнение матрицы змейкой

"""
Создаем матрицу-змейку
 
#n - размерность матрицы n x n
#b - результирующая матрица
#st - текущее значение-счетчик для записи в матрицу
#m - коеффициент, используемый для заполнения верхней
#матрицы последующих витков, т.к. одномерные матрицы
#следующих витков имеют меньше значений
"""
 
n = int(input("Размерность матрицы: "))
 
b = [[0]*n for i in range(n)]
st, m = 1, 0
# Заранее присваиваю значение центральному элементу
# матрицы
b[n//2][n//2]=n*n
for v in range(n//2):
    #Заполнение верхней горизонтальной матрицы
    for i in range(n-m):
        b[v][i+v] = st
        st+=1
    
    #Заполнение правой вертикальной матрицы    
    for i in range(v+1, n-v):
        b[i][-v-1] = st
        st+=1
        
    #Заполнение нижней горизонтальной матрицы
    for i in range(v+1, n-v):
        b[-v-1][-i-1] =st
        st+=1
        
    #Заполнение левой вертикальной матрицы
    for i in range(v+1, n-(v+1)):
        b[-i-1][v]=st
        st+=1
        
    m+=2
    
b.reverse()
for el in b:
    el.reverse()
    print(el)

n, m = map(int, input().split())
a = []
w = -1
a2 = []
for i in range(n):
    a1 = []
    for j in range(m):
        w = w + 1
        a1.append(w)
 
    if i % 2 == 1:
        a1.reverse()
        a.append(a1)
    else:
        a.append(a1)
 
for i in a:
    for j in range(m):
        print('{:<3d}'.format(i[j]), end='')
    print()

n = 10
# вместо range(1,n*n+1) вставить 
# отсортированный список (его наименование)
w = (i for i in range(1,n*n+1))
b = [[next(w)]]
 
for i in range(2*(n-1)):
    b = list(map(list,zip(*b)))
    b.reverse()
    for h in range(len(b)):
        b[h].append(next(w))
 
b = list(map(list,zip(*b)))
b.reverse()
b = list(map(list,zip(*b)))
for i in b:
        print(*i)
print()