numpy 如何取消循环操作

[going2think]

有如下代码，请问该如何去掉两重 for 循环，使用 numpy 的数组操作来实现呢？谢谢各位了

import numpy as np

a = np.zeros((100, 100))
b = np.zeros((100, 100))
ds = []
for i in range(5, 95):
    for j in range(5, 95):
        pa = a[i-5:i+5, j-5:j+5]
        pb = b[i-5:i+5, j-5:j+5]
        d = pa - pb
        ds.append(d)

return ds

[necomancer]

from skimage.util.shape import view_as_windows

a = np.random.random((100,100))
b = np.random.random((100,100))

ret = view_as_windows(a-b, (10,10))

In [1]: np.allclose(ret[:-1,:-1,:], ds.reshape(90,90,10,10))
Out[1]: True

顺带一提，view_as_windows 有 step 函数，你这个情况 step=1，是默认参数。

[necomancer]

step=10 在你的例子里应该相当于刚好无重叠，可以用 view_as_blocks 函数。

[necomancer]

这个函数基本上是在用 np.lib.stride_tricks.as_strided，所以如果你没有 skimage，去 skimage 找到那个函数拷过来就行。麻烦的地方只在怎么根据 window_size 和 step 算 stride

[going2think]

@necomancer 感谢帮助，我去试试～

[xgdgsc]

这种直接改 numba