python - scipy.ndimage.filters.convolve - 沿不同轴的不同模式？

Question

scipy.ndimage.filters 中的几个函数, 包括 scipy.ndimage.filters.convolve ，有一个“模式”参数，定义它在边界处的行为方式。 mode='constant' 对超出边界的点使用常量值，而 mode='wrap' 环绕。这适用于所有轴。

我想对二维数组进行卷积(例如)，以便:

• 边界外的轴 0 点环绕
• 轴1在边界外的点是常量

执行此操作最有效的方法是什么？

Answer 1

我可以使用 mode='wrap' 并在我希望保持不变的轴末端添加一些死角:

import numpy
from scipy import misc, ndimage

lena = misc.lena()
image = numpy.vstack((lena, numpy.zeros(lena.shape[1])))
weights = numpy.array([[1, 1, 1],
                    [1, 8, 1],
                    [1, 1, 1]])/16.
convimage = ndimage.convolve(image, weights, mode='wrap')[0:lena.shape[1],]