Задача 3. Диагональная матрица (Python)

Напишите программу, которая получает на вход размер квадратной матрицы и выводит на экран по такому принципу диагональ из единиц с левого нижнего угла до верхнего правого, ниже диагонали — двойки, выше — нули.

Пример:

Решение задачи:

size = int(input('Введите размер матрицы: '))
for row in range(size):
    for col in range(size):
        if col + row == size -1:
            print(1, end=' ')
        elif row >= size - col:
            print(2, end=' ')
        else:
            print(0, end=' ')
    print()

Проверяем: