將矩陣轉換爲對角矩陣

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