diff --git a/plotting.py b/plotting.py index 4e4f1ceba92b30e9ed00ceef96422bc0eaa6c950..4f55bfdbe34828b812361a15451745162b5fbfe2 100644 --- a/plotting.py +++ b/plotting.py @@ -223,8 +223,10 @@ def plot_tune_spectrum(ax, libera_data, xy, turn_range=None, time_range=None, tu if smoothing is not None: ls, = ax.plot(freq, mag, **dict(kwargs, zorder=-100, alpha=0.1)) if 'c' not in kwargs and 'color' not in kwargs: kwargs.update(c=ls.get_color()) - freq, mag, phase = avg(freq, mag, phase, n=smoothing) - + add_scale(ax, smoothing*np.mean(np.diff(freq))) + #freq, mag, phase = avg(freq, mag, phase, n=smoothing) + freq, mag, phase = [scipy.signal.savgol_filter(_, smoothing, 0) for _ in (freq, mag, phase)] # savgol filter with order 0 is moving average + ax.plot(freq, mag, **kwargs) ax.set(xlim=tune_range, xlabel=f'Tune $q_{xy}$',