将矩阵转换为对角矩阵

Question

我有一个向量，我需要将其转换为对角矩阵。我的矢量是如下

>>> mymatrix 
    matrix([[36, 66, 96]]) 

预计产量

36 0 0 
0 66 0 
0 0 96 

我试图np.diag(mymatrix)但它仅给出了[[36]]结果。我怎样才能做到这一点？

Answer 1

您需要diagflat因为你的输入是一个矩阵：

np.diagflat(mymatrix) 
Out: 
matrix([[36, 0, 0], 
     [ 0, 66, 0], 
     [ 0, 0, 96]])