2016-09-21 170 views
-1

我想创建一个矩阵,其中的矩阵是矩阵元素的对角元素其他?没有在Matlab矩阵的矩阵对角元素

A=[1 2;3 4] 
B=[1 0 0 0;0 2 0 0;0 0 3 0;0 0 0 4] 

回答

0

环在Matlab中的情况下,只是下面的

v = A'(:) % transpose since it makes long column by concat left to right column 
B = diag(v) 

在Python的情况下,你可以使用numpy的

import numpy as np 
A = np.array([[1,2], [3,4]]) 
B = np.diag(A.ravel())