Олимпиадный тренинг

Задача 37531. Diagonal Array


Задача

Темы: NumPy
The input is the number n. Output an array of size nxn, in in the diagonal numbers are 0 to n-1, and the rest of the numbers are 0.
 

 

Examples
# Input Output
1 5 [[0 0 0 0 0]
 [0 1 0 0 0]
 [0 0 2 0 0]
 [0 0 0 3 0]
 [0 0 0 0 4]]

Запрещенные операторы:if;for;while;do