1

我正在尝试使用函数设置信号的包络obspy.signal.filter.envelope。到目前为止，我的程序正在生成 2 行：红色的是原始信号。蓝色的是用带通滤波器（filter_sinc函数）过滤的信号。

现在，我想要红色信号的包络。但是，当我运行代码时，假设是红色信号包络的绿线不是，我不明白为什么。

此外，我想制作这些线条的平滑动画，以显示它们随着时间的推移而生成，我知道这不是一个很好的结果。我怎样才能让它运行得更好？喜欢显示平滑生成的线条吗？

这是我的 test.dat 文件： 文件

和代码：

from __future__ import division
import matplotlib.pyplot as plt
import matplotlib.animation as anim
import numpy as np
import obspy.signal as sign

def init():
    im.set_data([], [])
    im2.set_data([], [])
    im3.set_data([], [])
    return im,im2,

def filter_sinc(x=None,n=None):
    f=1 / n
    nf=2 * np.fix(2 / f)
    o=(np.arange(1,nf+1)) - nf / 2
    filt=0 * o + 2 * f    
    temp = np.sin([2 * np.pi * f * np.concatenate((o[0:nf/2-1],o[nf/2:]),1)]) / np.concatenate((o[0:nf/2-1],o[nf/2:]),1) / np.pi
    filt[0:nf/2-2] = temp[:,0:nf/2-2]
    filt[nf/2:] = temp[:,nf/2-1:]   
    hamm=0.54 - 0.46 * np.cos(2 * np.pi * (np.arange(1,nf+1)) / nf)
    filt=filt*hamm
    xf = np.convolve(x,filt,mode='same')
    return xf


fig = plt.figure()
ax2 = fig.add_subplot(1,1,1)    

filtre_x = []
filtre_y = []
mesure = []
temps = []
hilb = []

i=0
n=30
zoom=100

fichier_dat = np.genfromtxt("test.dat", usecols = (0,1,2,3,4,5,6), comments = "#")
mesure = fichier_dat[:,6]
temps = fichier_dat[:,2]*86400 + fichier_dat[:,3]*3600 + fichier_dat[:,4]*60 + fichier_dat[:,5]

hilb = sign.filter.envelope(mesure)

while i < np.size(temps,0)-1200:
    filtre_x = np.hstack([filtre_x,temps[np.arange(i+n,i+(1200-n))]])
    filtre_y = np.hstack([filtre_y,filter_sinc(mesure[np.arange(i,i+1200)],n*2)[np.arange(n,1200-n)]])
    i = i+1200

filtre_x = filtre_x/20
temps = temps/20

ax2.set_xlim([filtre_x[0],filtre_x[0]+zoom])
ax2.set_ylim([0,6])

im, = ax2.plot([], [], color=(1,0,0))
im2, = ax2.plot([], [], color=(0,0,1))
im3, = ax2.plot([], [], color=(0,1,0))


def animate(k):

    im.set_xdata(temps)
    im.set_ydata(mesure)

    im2.set_xdata(filtre_x)
    im2.set_ydata(filtre_y)

    im3.set_xdata(temps)
    im3.set_ydata(hilb)

    ax2.set_xlim(k+filtre_x[0], k+filtre_x[0]+zoom)


ani = anim.FuncAnimation(fig, animate, init_func=init, frames=np.size(filtre_x), interval=30, blit=False)
plt.show()
4

0 回答 0