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

Задача 37531. Диагональный массив


Задача

Темы: NumPy
На вход подаётся число n. Выведите массив размера nхn, в котором по диагонали идут числа 0 до n-1, а остальные числа равны 0.
 

 

Примеры
Входные данные Выходные данные
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