Создать матрицу размером N×N и заполнить ее следующим образом: элементы на главной диагонали равны 1, выше диагонали — 2, ниже — 0. Размера матрицы вводится пользователем.
N = int(input()) # вводим размер матрицы
matrix = [[0 for x in range(N)] for y in range(N)] # создаем матрицу и заполняем нулями
for i in range(N): # заполняем диагональ единицами
matrix[i][i] = 1
for i in range(N): # заполняем элементы выше диагонали двойками
for j in range(i + 1, N):
matrix[i][j] = 2
# Выводим получившуюся матрицу
for row in matrix:
print(row)
