Заполнение выше главной диагонали

Создать матрицу размером 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)




Добавить комментарий

;-) :| :x :twisted: :smile: :shock: :sad: :roll: :razz: :oops: :o :mrgreen: :lol: :idea: :grin: :evil: :cry: :cool: :arrow: :???: :?: :!: