Условие

Дано число n. Создайте массив размером n×n и заполните его по следующему правилу:

Числа на диагонали, идущей из правого верхнего в левый нижний угол равны 1.

Числа, стоящие выше этой диагонали, равны 0.

Числа, стоящие ниже этой диагонали, равны 2.

Полученный массив выведите на экран. Числа в строке разделяйте одним пробелом.

Решение

1. n = int(input())
2. a = [[0] * n for i in range(n)]
3. for i in range(n):
4.     a[i][n - i - 1] = 1
5. for i in range(n):
6.     for j in range(n - i, n):
7.         a[i][j] = 2
8. for row in a:
9.     for elem in row:
10.         print(elem, end=' ')
11.     print()