python - Vstack 获取数组

Question

我试图将我从 z 得到的向量放在 A 中，从 q 得到的向量放在 B 中，然后将它们堆叠在一起。arr 是大小为 (600,478) 的图像的数组。M=600，N=478。由于某种原因，我得到的矩阵不是正确的。

def avgdif(M):
    A=np.zeros([int(M/2),N])
    B=np.zeros([int(M/2),N])
    for i in range(int(M/2)):
        z=(np.add(arr[2*i-1],arr[2*i]))/2
        A[i:]=[z] 
        q=(np.subtract(arr[2*i-1],arr[2*i]))/2
        B[i:]=[q]
        
        
        
        
    return np.vstack([A,B])