python: 1D plot + smoothing

In [61]:

import numpy as np
import matplotlib.pyplot as plt
import scipy
from scipy.signal import savgol_filter

def fun(t, a, b, tau):
    return a * np.sin(b * t) * np.exp(- t * tau)

x = np.arange(200)
ra = 0.05 * np.random.randn(200)
y = fun(x, 1, 0.04, 0.01)+ ra

plt.plot(x, y, marker = 'o', linestyle = '', markersize = 2,  label = 'raw data')
y_sf = savgol_filter(y, 25, 1)
plt.plot(x, y_sf, linewidth = 3, label = 'smoothed data')
plt.legend()

Out[61]:

<matplotlib.legend.Legend at 0x7fb4f017de50>