将矩阵转换为对角矩阵

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]])