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]])