1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
| import matplotlib.pyplot as plt import numpy as np
np.random.seed(0)
dt = 0.01 Fs = 1 / dt t = np.arange(0, 10, dt)
nse = np.random.randn(len(t)) r = np.exp(-t / 0.05) cnse = np.convolve(nse, r) * dt cnse = cnse[:len(t)]
s = 0.1 * np.sin(4 * np.pi * t) + cnse
fig, axes = plt.subplots(nrows=3, ncols=2, figsize=(7, 7))
axes[0, 0].set_title("Signal") axes[0, 0].plot(t, s, color='C0') axes[0, 0].set_xlabel("Time") axes[0, 0].set_ylabel("Amplitude")
axes[1, 0].set_title("Magnitude Spectrum") axes[1, 0].magnitude_spectrum(s, Fs=Fs, color='C1')
axes[1, 1].set_title("Log. Magnitude Spectrum") axes[1, 1].magnitude_spectrum(s, Fs=Fs, scale='dB', color='C1')
axes[2, 0].set_title("Phase Spectrum ") axes[2, 0].phase_spectrum(s, Fs=Fs, color='C2')
axes[2, 1].set_title("Angle Spectrum") axes[2, 1].angle_spectrum(s, Fs=Fs, color='C2')
axes[0, 1].remove()
fig.tight_layout() plt.show()
|