python中的循环卷积

时间:2016-05-10 23:38:38

标签: python numpy

Python是否有办法在两个1D阵列之间执行循环卷积, 喜欢Matlab函数cconv? 我试过numpy.convolve,但它不是一样的,我找不到相应的。

1 个答案:

答案 0 :(得分:3)

您可以尝试以下命令:scipy.ndimage.filters.convolve1d

您有一个名为mode的选项,您可以编写mode = wrap

使用它,您可以获得周期性边界条件作为卷积的填充

例如:

result = scipy.ndimage.convolve(image,kernel,mode='wrap')

import numpy as np
image = np.array([[0, 0, 0, 0],
                  [0, 0, 0, 1],
                  [0, 0, 0, 0]])
kernel = np.array([[1, 1, 1],
                    [1, 1, 1],
                    [1, 1, 1]])
from scipy.ndimage import convolve
convolve(image, kernel, mode='wrap')
array([[1, 0, 1, 1],   
       [1, 0, 1, 1],
       [1, 0, 1, 1]])