DenoiseSignal.py

# tuotetaan kohinainen signaali
# poistetaan kohinaa IIR-suodattimen avulla (lfilter)
# Juhani Kaukoranta 26.1.2024
import matplotlib.pyplot as plt
import numpy as np
from scipy.signal import lfilter

x = np.linspace(0,np.pi , 1000)  # x axis
y = np.sin(x**2) * np.cos(x) # alkuperäinen signaali
noise = np.std(y)*np.random.standard_normal(len(x))/6 # kohinaa
y = y + noise # kohina lisätty signaaliin

# suodatin
n = 50  # mitä suurempi n, suurempi suodatus
b = [1.0 / n] * n
a = 1
yy = lfilter(b, a, y)

# käyrien (2 kpl) tulostus
plt.figure()
plt.subplot(211)
plt.title("yllä kohinainen signaali, alla suodatettu signaali")
plt.plot(x, y, linewidth=1, linestyle="-", c="b")  # kohinainen signaali

plt.subplot(212)
plt.plot(x, yy, linewidth=2, linestyle="-", c="b")  # suodatettu kohinaa pois
plt.show()