I'm trying to convert my working 2nd-order Butterworth low pass filter to 1st-order in python, but it gives me very big numbers, like flt_y_1st[299]: 26198491071387576370322954146679741443295686950912.0. Here's my 2nd-order and 1st-order Butterworth:
import math
import numpy as np
import matplotlib.pyplot as plt
from scipy.signal import lfilter
from scipy.signal import butter
def butter_lowpass(cutoff, fs, order=1):
nyq = 0.5 * fs
normal_cutoff = cutoff / nyq
b, a = butter(order, normal_cutoff, btype='low', analog=False)
return b, a
def butter_lowpass_filter(data, cutoff, fs, order=1):
b, a = butter_lowpass(cutoff, fs, order=order)
y = lfilter(b, a, data)
return y
def bw_2nd(y, fc, fs):
filtered_y = np.zeros(len(y))
omega_c = math.tan(np.pi*fc/fs)
k1 = np.sqrt(2)*omega_c
k2 = omega_c**2
a0 = k2/(1+k1+k2)
a1 = 2*a0
a2 = a0
k3 = 2*a0/k2
b1 = -(-2*a0+k3)
b2 = -(1-2*a0-k3)
filtered_y[0] = y[0]
filtered_y[1] = y[1]
for i in range(2, len(y)):
filtered_y[i] = a0*y[i]+a1*y[i-1]+a2*y[i-2]-(b1*filtered_y[i-1]+b2*filtered_y[i-2])
return filtered_y
def bw_1st(y, fc, fs):
filtered_y = np.zeros(len(y))
omega_c = math.tan(np.pi*fc/fs)
k1 = np.sqrt(2)*omega_c
k2 = omega_c**2
a0 = k2/(1+k1+k2)
a1 = 2*a0
k3 = 2*a0/k2
b1 = -(-2*a0+k3)
# b1 = -(-2*a0) # <= Removing k3 makes better, but still not perfect
filtered_y[0] = y[0]
for i in range(1, len(y)):
filtered_y[i] = a0*y[i]+a1*y[i-1]-(b1*filtered_y[i-1])
return filtered_y
f = 100
fs = 2000
x = np.arange(300)
y = np.sin(2 * np.pi * f * x / fs)
flt_y_2nd = bw_2nd(y, 120, 2000)
flt_y_scipy = butter_lowpass_filter(y, 120, 2000, 1)
flt_y_1st = bw_1st(y, 120, 2000)
for i in x:
print('y[%d]: %6.3f flt_y_2nd[%d]: %6.3f flt_y_scipy[%d]: %6.3f flt_y_1st[%d]: %8.5f' % (i, y[i], i, flt_y_2nd[i], i, flt_y_scipy[i], i, flt_y_1st[i]))
plt.subplot(1, 1, 1)
plt.xlabel('Time [ms]')
plt.ylabel('Acceleration [g]')
lines = plt.plot(x, y, x, flt_y_2nd, x, flt_y_scipy, x, flt_y_1st)
l1, l2, l3, l4 = lines
plt.setp(l1, linewidth=1, color='g', linestyle='-')
plt.setp(l2, linewidth=1, color='b', linestyle='-')
plt.setp(l3, linewidth=1, color='y', linestyle='-')
plt.setp(l4, linewidth=1, color='r', linestyle='-')
plt.legend(["y", "flt_y_2nd", "flt_y_scipy", "flt_y_1st"])
plt.grid(True)
plt.xlim(0, 150)
plt.ylim(-1.5, 1.5)
plt.title('flt_y_2nd vs. flt_y_scipy vs. flt_y_1st')
plt.show()
... I removed all [i-2]s, which were a feed-forward and a feed-back.
However, it seems that's not enough. I think I need to change some equations in a0, b1, etc. For example, when I remove '+k3' from b1, I get a plot like this (looks better, doesn't it?):
I'm not specialized in filters, but at least I know that this 1st order differs from that of scipy.butter. So, please help me find correct coefficients. Thank you in advance.
Here's my reference: filtering_considerations.pdf
Let me answer to myself.
The final coefficients are:
omega_c = math.tan(np.pi*fc/fs)
k1 = np.sqrt(2)*omega_c
a0 = k1/(math.sqrt(2)+k1)
a1 = a0
b1 = -(1-2*a0)
Here's how. I reverse-engineered them from scipy.butter, as @sizzzzlerz suggested (thanks). scipy.butter spits out these coefficients:
b: [ 0.16020035 0.16020035]
a: [ 1. -0.6795993]
Note that b and a are reversed from my reference. They're gonna be:
a0 = 0.16020035
a1 = 0.16020035
b0 = 1
b1 = -0.6795993
Then, applied these coefficients to my incomplete formula:
a1 = a0 = 0.16020035
b1 = -(1-2*a0) = -{1-2*(0.16020035)} = -(0.6795993)
So far, so good. Incidentally:
k1 = 0.2698
k2 = 0.0364
So:
a0 = k2/(1+k1+k2) = 0.0364/(1+0.2698+0.0364) = 0.0279
... This is far from 0.16020035. At this point, I eliminated k2 and put like this:
a0 = k1/(1+k1+x)
When x = 0.4142, I got 0.16020164. Close enough.
a0 = k1/(1+k1+0.4142) = k1/(1.4142+k1)
... 1.4142 ...!? I've ever seen this number before ...:
= k1/(math.sqrt(2)+k1)
Now the plot looks like this (flt_y_scipy is completely covered by flt_y_1st):
You can search for the keywords "first order" butterworth "low pass filter" "sqrt(2)", etc.
This is the end of my Sunday DIY. ;-)