Заполнение диагоналей в матрице

Заполнить квадратную матрицу таким образом, чтобы на главной и побочной диагоналях были 1, остальные элементы равны 0. Размер предварительно вводится пользователем.

n = int(input("Введите размер матрицы: "))
matrix = [[0] * n for i in range(n)]
for i in range(n):
    for j in range(n):
        if i == j or i + j == n- 1:
            matrix[i][j] = 1
for row in matrix:
    print(*row)




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

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