fir.py 文件源码

def transform(self, sigin):
        """Apply this filter to a signal

        Parameters
        ----------
        sigin : array, shape (n_points, ) or (n_signals, n_points)
            Input signal

        Returns
        -------
        filtered : array, shape (n_points, ) or (n_signals, n_points)
            Filtered signal
        """
        sigin_ndim = sigin.ndim
        sigin = np.atleast_2d(sigin)
        filtered = [signal.fftconvolve(sig, self.fir, 'same') for sig in sigin]

        if sigin_ndim == 1:
            filtered = filtered[0]
        else:
            filtered = np.asarray(filtered)

        return filtered