segypy.py 文件源码

def wiggle(Data,SH,skipt=1,maxval=8,lwidth=.1):
    """
    wiggle(Data,SH)
    """
    import matplotlib.pylab as plt

    t = range(SH['ns'])
#     t = range(SH['ns'])*SH['dt']/1000000;

    for i in range(0,SH['ntraces'],skipt):
#        trace=zeros(SH['ns']+2)
#        dtrace=Data[:,i]
#        trace[1:SH['ns']]=Data[:,i]
#        trace[SH['ns']+1]=0
        trace=Data[:,i]
        trace[0]=0
        trace[SH['ns']-1]=0    
        plt.plot(i+trace/maxval,t,color='black',linewidth=lwidth)
        for a in range(len(trace)):
            if (trace[a]<0):
                trace[a]=0;
        # pylab.fill(i+Data[:,i]/maxval,t,color='k',facecolor='g')
        plt.fill(i+Data[:,i]/maxval,t,'k',linewidth=0)
    plt.title(SH['filename'])
    plt.grid(True)
    plt.show()

#%%