Диагонализировать несколько векторов, используя Numpy ⇐ Python

[Anonymous]

Скажем, у меня есть матрица формы (2,3), мне нужно диагонализировать вектор 3-элементов на матрицу формы (3,3), для всех 2 векторов одновременно. То есть мне нужно вернуть матрицу с формой (2,3,3). Как я могу сделать это с элегантным Numpy? < /P>

Код: Выделить всё

given data = np.array([[1,2,3],[4,5,6]])

i want the result [[[1,0,0],
[0,2,0],
[0,0,3]],

[[4,0,0],
[0,5,0],
[0,0,6]]]

Спасибо

Подробнее здесь: https://stackoverflow.com/questions/737 ... sing-numpy