Solve problem "The diagonal parallel to the main" online - Learn Python 3 - Snakify

The diagonal parallel to the main

Statement

Given an integer $$n$$, produce a two-dimensional array of size $$\left ( n \times n \right )$$ and complete it according to the following rules, and print with a single space between characters:
• On the main diagonal write  0 .
• On the diagonals adjacent to the main, write  1 .
• On the next adjacent diagonals write  2  and so forth.

Print the elements of the resulting array.

In all the problems input the data using input() and print the result using print().