adds axis keyword argument to smooth

nirum · May 1, 2016 · a284b4a · a284b4a
1 parent 55706c6
commit a284b4a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/jetpack/signals.py b/jetpack/signals.py
@@ -132,7 +132,7 @@ def xcorr(x, y, maxlag):
     return lags, corr
 
 
-def smooth(x, sigma=1.0):
+def smooth(x, sigma=1.0, axis=0):
     """
     Smooths a 1D signal with a gaussian filter
 
@@ -148,10 +148,8 @@ def smooth(x, sigma=1.0):
     -------
     xs : array_like
         A smoothed version of the input signal
-
     """
-
-    return gaussian_filter1d(x, sigma, axis=0)
+    return gaussian_filter1d(x, sigma, axis=axis)
 
 
 def norms(x, order=2):