计算离散卷积

convelution.py

import numpy as np

def conv_m(x, nx, h, nh):
    '''计算x与h的卷积，nx与nh为其序列下标范围'''
    nyb = nx[0] + nh[0]
    nye = nx[-1] + nh[-1]
    ny = np.arange(nyb, nye+1)
    y = np.convolve(x, h)
    return y, ny