Python的矩陣重疊

Question

我有一個問題，重疊矩陣

import numpy as np 
import pprint 

A = np.array([[1,0],[0,1]]) 
B = np.array([[0,0,0],[0,0,0],[0,0,0]]) 

for i in range(0,2): 
    for j in range(0,2): 
     if i==0: 
      if j<2: 
       B[i,j]=A[i,j] 
     elif i==2: 
      if j==2: 
       B[i,j]=A[i,j] 
     else: 
      B[i,j] = A[i,j] + A[i-1,j-1] 

pprint.pprint(B) 

，並返回數組這樣([[1,0,0],[0,2,0],[0,0,0]])，但我想([[1,0,0],[0,2,0],[0,0,1]])

什麼是錯我的代碼？

請大家幫忙。

Answer 1

你可以不用任何的循環：

B[:2, :2] += A 
B[1:3, 1:3] += A 
print(B) 
# array([[1, 0, 0], 
#  [0, 2, 0], 
#  [0, 0, 1]])