print(*reversed(i))
n = int(input())
l = [[0 for j in range(n)] for i in range(n)]
i = 0
j = 0
k = 1
while k <= n ** 2:
l[i][j] = k
if i <= j + 1 and i + j < n - 1:
j += 1
elif i < j and i + j >= n-1:
i += 1
elif i >= j and i + j > n-1:
j -= 1
else:
i -= 1
k += 1
for i in l:
print(*i)
Теперь программа будет требовать от пользователя ввода до тех пор, пока он не введет верный символ операции (новые символы операций добавляй в список с ними, все легко)