Python 实现 Savitzky - Golay 滤波的详细过程及注意要点

以下是一个使用 Python 实现 Savitzky-Golay 滤波的示例代码： ```python import numpy as np def savitzky_golay(y, window_size, order): # 计算窗口内的系数 half_window = (window_size - 1) // 2 m = np.arange(-half_window, half_window + 1) b = np.polyfit(m, np.ones(window_size), order) a = np.polyval(b, m) a = a / np.sum(a) # 进行滤波 y_padded = np.pad(y, (half_window, half_window), mode='edge') y_filtered = np.convolve(y_padded, a, mode='valid') return y_filtered # 示例用法 data = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) filtered_data = savitzky_golay(data, 5, 2) print(filtered_data) ``` 额外需要注意的地方： 1. 窗口大小和阶数的选择会影响滤波效果，需要根据实际数据进行调整。 2. 对于边缘数据的处理方式，需要根据具体需求选择合适的填充方法。 3. 在实际应用中，可能需要对数据进行预处理，例如去除异常值或进行标准化。 ######[AI写代码神器 | 319点数解答 | 2024-12-20 14:16:01]