Python. Задача 5-10

Пользователь вводит натуральное число N < 20. Создать таблицу, в которой на главной диагонали записаны нули, на двух прилегающих диагоналях  с двух сторон к главной записаны 1, на следующих - 2 и т.д. Вывести таблицу.

Решение
a = []
N = int(input('Введите натуральное число N < 20: '))
# Создаем список из N строк, в каждой список из N нулей
a = [[0] * N for i in range(N)]
# Заполняем таблицу
for i in range(N):
    for j in range(N):
        a[i][j] = abs(j - i)
# Выводим таблицу
for i in range(N):
    for j in range(N):
        print('%3d' % a[i][j], end = '')
    print()