From 3a5e4fe8f63240e5c4ea64ea369922a64a2208ce Mon Sep 17 00:00:00 2001 From: eryk-urbanski Date: Wed, 10 Apr 2024 13:44:57 +0200 Subject: [PATCH] Replace np.append method with scipy.signal.lfilter --- spafe/utils/preprocessing.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/spafe/utils/preprocessing.py b/spafe/utils/preprocessing.py index b5bd231..0b2517a 100644 --- a/spafe/utils/preprocessing.py +++ b/spafe/utils/preprocessing.py @@ -6,9 +6,11 @@ For a copy, see . """ + from typing import Tuple import numpy as np +from scipy.signal import lfilter from dataclasses import dataclass from typing_extensions import Literal @@ -67,7 +69,7 @@ def pre_emphasis(sig: np.ndarray, pre_emph_coeff: float = 0.97) -> np.ndarray: y[t] = x[t] - \\alpha \\times x[t-1] """ - return np.append(sig[0], sig[1:] - pre_emph_coeff * sig[:-1]) + return lfilter([1, -pre_emph_coeff], [1], sig) def stride_trick(a: np.ndarray, stride_length: int, stride_step: int) -> np.ndarray: