行列の行列対角要素

Question

行列を作成したいところここで、行列は行列要素の対角要素です。 Matlabの場合はMatlabの

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

Answer 1

のループせずに、単にPythonの場合

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

を次の操作を行い、あなたは

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

をnumpyの使用することができます