signal_models.py 文件源码

def plot(self):
        """ Plot a realisation of the signal waveform """
        Y=self.rvs()
        Y_processed=linear_transform(Y,self.preprocessing_method)
        N,L=Y_processed.shape

        if ((L==3) or (L==1)):
            n_vect=np.arange(N)/self.Fe
            for l in range(L):
                plt.plot(n_vect,Y_processed[:,l],label="signal %d" %l)
            plt.xlabel("Time")
            plt.ylabel("Signal")
            plt.legend()

        if L==2:
            z=Y_processed[:,0]+1j*Y_processed[:,1]
            plt.plot(np.real(z),np.imag(z))
            plt.xlabel("Real Part")
            plt.ylabel("Imag Part")