两个 for 循环的向量化

Question

我正在尝试对一个函数进行矢量化，而且我快到了。剩下的就是这个

for x in range(0,img.shape[1]):
        for y in range(0,img.shape[0]):  

            if w00[y,x] > 0:
                total[y,x] += img.item((y0[y,x], x0[y,x])) * w00[y,x]
            if w01[y,x] > 0:
                total[y,x] += img.item((y1[y,x], x0[y,x])) * w01[y,x]
            if w10[y,x] > 0:
                total[y,x] += img.item((y0[y,x], x1[y,x])) * w10[y,x]
            if w11[y,x] > 0:
                total[y,x] += img.item((y1[y,x], x1[y,x])) * w11[y,x]

现在看起来很丑，因为我已经对其他所有内容进行了矢量化，并引入了这个 for 循环以不破坏代码。当索引每次切换时，我很难对这部分进行矢量化。一些帮助会很好。

y1 和 x1 也就是 x0 + 1 和 y0 + 1，这样可以简化向量化

Answer 1

这样的事情会起作用

for w, x, y in zip([w00,w01,w10,w11], [y0,y1,y0,y1], [x0,x0,x1,x1]):
    total += img[y,x] * w * (w>0)

编辑：可能需要进行一些修改。由于缺少示例数据，我无法测试代码是否正常运行。