Помощь в задаче Питон

Question

На вход программе подается натуральное число n. Напишите программу, которая создает матрицу размером n×n и заполняет её по следующему правилу:
*на главной диагонали на месте каждого элемента должно стоять число 0; 
*на двух диагоналях, прилегающих к главной, число 1; 
*на следующих двух диагоналях число 2, и т.д.

leirby · Accepted Answer

n = int(input()) 
s = [['.' for x in range(n)] for x in range(n)] 
l = [int(x) for x in range(1,n)] 
for i in range(n): 
    for j in range(n): 
        if i==j: 
            s[i][j] = 0 
            for t in range(len(l)): 
                if j+l[t]<=n-1: 
                    s[i][j+l[t]]=l[t] 
                if 0<=j-l[t]<=n-1: 
                    s[i][j-l[t]]=l[t] 
for x in s:print(*x)