2017-06-25 8 views
1

私はベクトルを持っており、対角行列に変換する必要があります。行列を対角行列に変換する

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

の予想される出力は、私がnp.diag(mymatrix)を試みたが、それだけで[[36]]結果を与える

36 0 0 
0 66 0 
0 0 96 

で次のように私のベクトルです。これどうやってするの?

答えて

5

あなたの入力は行列であるので、あなたはdiagflatが必要になります。

np.diagflat(mymatrix) 
Out: 
matrix([[36, 0, 0], 
     [ 0, 66, 0], 
     [ 0, 0, 96]]) 
関連する問題