有如下代码，请问该如何去掉两重 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  | 1 necomancer Jun 6, 2019  1 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，是默认参数。 |
| 2 necomancer Jun 6, 2019 1 step=10 在你的例子里应该相当于刚好无重叠，可以用 view_as_blocks 函数。 |
| 3 necomancer Jun 6, 2019 1 这个函数基本上是在用 np.lib.stride_tricks.as_strided，所以如果你没有 skimage，去 skimage 找到那个函数拷过来就行。麻烦的地方只在怎么根据 window_size 和 step 算 stride |
 | 4 going2think OP @necomancer 感谢帮助，我去试试～ |
 | 5 xgdgsc Jun 7, 2019 这种直接改 numba |
Select Language