Python生成ricker子波（瑞克子波）的函数

import numpy as np
def ricker(f, dt):

#Ricker子波的函数
    nw = 2.2/f/dt
    nw = 2 * floor(nw/2)+2
    nc = floor(nw/2)
    w = np.zeros(nw + 1,).reshape(-1, 1)
    k = np.array(list(i+1 for i in range(1, nw))).reshape(1, -1).T
    alpha = (nc - k + 1) * f * dt * np.pi
    beta = alpha ** 2
    w = (1 - beta * 2) * np.exp(-beta)
    
    return w