python实现数据平滑函数

Python中有多种方法可以实现数据平滑，其中最常用的是移动平均法、指数移动平均法（Exponential Moving Average, EMA）、Savitzky-Golay滤波等。这里简单介绍两种常见的：

1. **移动平均法**：通过计算一段时间内数据的算术平均值来进行平滑。例如，你可以使用pandas库中的`rolling()`函数：

import pandas as pd

def smooth_data(data, window_size):
    return data.rolling(window=window_size).mean()

data = [1, 2, 3, 4, 5, 6, 7, 8, 9]
smoothed_data = smooth_data(pd.Series(data), 3)

2. **Savitzky-Golay滤波**：这是一种基于多项式拟合的方法，适用于处理噪声较多的数据。scipy库提供了一个实现：

from scipy.signal import savgol_filter

def smooth_with_savgol(data, window_length, polyorder):
    return savgol_filter(data, window_length, polyorder)

simplified_data = smooth_with_savgol(data, 5, 2)