Numpy 卷积

numpy.convolve 文档

import numpy as np

>>> a = [1,2,1] # 卷积核(对称的)
>>> b = [1,3,5,6,7]
>>> np.convolve(b,a,"valid") # 只做有效的运算，第一个窗是[1,3,5]，窗内元素均有效
array([12, 19, 24])
>>> np.convolve(b,a,"full") # 全卷积，第一个窗是[Nan,Nan,1],窗内有元素即可
array([ 1,  5, 12, 19, 24, 20,  7])
>>> np.convolve(b,a,"same") # 同维卷积，第一个窗是[Nan,1,5]，窗中心包含元素即可
array([ 5, 12, 19, 24, 20])

Numpy 窗函数