NumPy: создайте трехмерный массив с единицами по диагонали и нулями в других местах

NumPy: Array Object Exercise-42 с решением

Напишите программу NumPy для создания трехмерного массива с единицами по диагонали и нулями в других местах.

Иллюстрированная презентация:

Пример решения : -

Код Python:

import numpy as np
x = np.eye(3)
print(x)

Пример вывода:

 [[1. 0. 0.]                                                         
 [0. 1. 0.]                                                         
 [0. 0. 1.]]

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу NumPy для создания массива из 10 с такой же формой и типом данного массива.
Далее: Напишите программу NumPy для создания двумерного массива, диагональ которого равна [4, 5, 6, 8] и 0 где-либо еще.