矩陣的矩陣對角元素

Question

我想創建一個矩陣，其中的矩陣是矩陣元素的對角元素其他？沒有在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

環在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())