Tạo mảng numpy khối chéo từ một mảng sumpy nhất định

Question

Tôi có một mảng numpy 2 chiều với số cột và hàng bằng nhau. Tôi muốn sắp xếp chúng thành một mảng lớn hơn có những cái nhỏ hơn trên đường chéo. Có thể chỉ định tần suất ma trận bắt đầu phải nằm trên đường chéo. Ví dụ:

a = numpy.array([[5, 7], 
       [6, 3]]) 

Vì vậy, nếu tôi muốn mảng này 2 lần trên đường chéo đầu ra mong muốn sẽ là:

array([[5, 7, 0, 0], 
     [6, 3, 0, 0], 
     [0, 0, 5, 7], 
     [0, 0, 6, 3]]) 

cho 3 lần:

array([[5, 7, 0, 0, 0, 0], 
     [6, 3, 0, 0, 0, 0], 
     [0, 0, 5, 7, 0, 0], 
     [0, 0, 6, 3, 0, 0], 
     [0, 0, 0, 0, 5, 7], 
     [0, 0, 0, 0, 6, 3]]) 

Có một cách nhanh chóng để thực hiện điều này với các phương thức gumpy và cho các kích thước tùy ý của mảng bắt đầu (vẫn xem xét mảng bắt đầu có cùng số hàng và cột)?

Answer 1

trường hợp cổ điển của numpy.kron -

np.kron(np.eye(n), a) 

mẫu chạy -

In [57]: n = 2 

In [58]: np.kron(np.eye(n), a) 
Out[58]: 
array([[ 5., 7., 0., 0.], 
     [ 6., 3., 0., 0.], 
     [ 0., 0., 5., 7.], 
     [ 0., 0., 6., 3.]]) 

In [59]: n = 3 

In [60]: np.kron(np.eye(n), a) 
Out[60]: 
array([[ 5., 7., 0., 0., 0., 0.], 
     [ 6., 3., 0., 0., 0., 0.], 
     [ 0., 0., 5., 7., 0., 0.], 
     [ 0., 0., 6., 3., 0., 0.], 
     [ 0., 0., 0., 0., 5., 7.], 
     [ 0., 0., 0., 0., 6., 3.]])