从numpy数组中去除方波状伪影

import numpy as np
from scipy import signal

# 生成包含方波状伪影的numpy数组
# 这里假设原始信号是一个正弦波
t = np.linspace(0, 1, 1000)
original_signal = np.sin(2 * np.pi * 5 * t)  # 5Hz的正弦波
sampling_error = np.random.normal(0, 0.1, 1000)  # 采样误差
noisy_signal = original_signal + sampling_error

# 设计一个低通滤波器
cutoff_freq = 10  # 截止频率为10Hz
nyquist_freq = 0.5  # 奈奎斯特频率
normalized_cutoff_freq = cutoff_freq / (0.5 * nyquist_freq)
b, a = signal.butter(4, normalized_cutoff_freq, 'low')

# 使用滤波器去除方波状伪影
filtered_signal = signal.lfilter(b, a, noisy_signal)

# 打印去除伪影后的信号
print(filtered_signal)