别再只用Excel了！用Python的NumPy和Pandas实现移动平均预测（附完整代码）

用Python实现时间序列预测从移动平均到实战优化在数据分析领域时间序列预测一直是个热门话题。传统工具如Excel虽然简单易用但当面对海量数据或复杂分析需求时就显得力不从心。Python生态中的NumPy和Pandas库提供了更强大的解决方案不仅能处理更大规模的数据还能实现更复杂的分析逻辑。本文将带你深入理解移动平均法的Python实现并分享几个提升预测准确性的实用技巧。1. 移动平均法基础与Python实现移动平均法是时间序列预测中最基础也最实用的方法之一。它的核心思想是通过计算近期数据的平均值来消除随机波动揭示数据的内在趋势。这种方法特别适合那些没有明显季节性和趋势变化的数据集。在Python中我们可以用Pandas的rolling方法轻松实现简单移动平均(SMA)import pandas as pd import numpy as np # 创建示例数据 data pd.Series([423, 358, 434, 445, 527, 429, 426, 502, 480, 384, 427, 446]) # 计算3期简单移动平均 sma_3 data.rolling(window3).mean() print(3期简单移动平均结果:) print(sma_3)这段代码中window3表示我们使用最近3期的数据进行平均计算。Pandas的rolling方法会自动处理边界条件前两个数据点因为没有足够的历史数据结果会是NaN。移动平均窗口大小的选择是个关键问题较小的窗口如3-5对数据变化反应灵敏但可能保留过多噪声较大的窗口如10-12平滑效果更好但可能掩盖重要趋势变化实际应用中可以通过计算不同窗口大小下的预测误差来选择最优参数def evaluate_sma(series, window): sma series.rolling(windowwindow).mean() mse ((series[window:] - sma[window:-1])**2).mean() return mse windows range(2, 7) for w in windows: print(f窗口大小{w}的MSE: {evaluate_sma(data, w):.2f})2. 加权移动平均与卷积实现简单移动平均给所有历史数据赋予相同的权重这在实际应用中可能不够理想。加权移动平均(WMA)通过给不同时期的数据分配不同权重可以更灵活地调整模型对近期数据的敏感度。NumPy的convolve函数是实现加权移动平均的利器def weighted_moving_average(series, weights): 计算加权移动平均 :param series: 输入序列 :param weights: 权重数组总和应为1 :return: 加权移动平均结果 return np.convolve(series, weights, modevalid) # 定义权重 - 近期数据权重更高 weights np.array([0.1, 0.2, 0.3, 0.4]) weights weights / weights.sum() # 确保权重总和为1 wma_result weighted_moving_average(data.values, weights) print(加权移动平均结果:) print(wma_result)这里有几个实用技巧值得注意权重数组的长度决定了移动平均的期数权重通常按从远到近递增以强调近期数据的重要性modevalid确保只返回完全重叠的部分避免边界效应对于更复杂的权重模式可以考虑使用指数递减权重# 指数递减权重 n 5 exp_weights np.exp(np.linspace(0, 1, n)) exp_weights exp_weights / exp_weights.sum() wma_exp weighted_moving_average(data.values, exp_weights)3. 处理真实世界数据的实用技巧现实中的数据很少像教科书示例那样干净整洁。以下是几个处理常见问题的实用方法3.1 处理缺失值移动平均对缺失值特别敏感。Pandas提供了多种填充方法# 前向填充 data_ffill data.fillna(methodffill) # 线性插值 data_interp data.interpolate(methodlinear) # 季节性插值 data_seasonal data.interpolate(methodtime)3.2 数据标准化当数据量纲差异大时标准化可以提高模型表现from sklearn.preprocessing import StandardScaler scaler StandardScaler() data_scaled scaler.fit_transform(data.values.reshape(-1, 1))3.3 异常值处理移动平均对异常值敏感可以使用滚动统计量识别和处理异常# 计算滚动Z-score rolling_mean data.rolling(window5).mean() rolling_std data.rolling(window5).std() z_scores (data - rolling_mean) / rolling_std # 标记绝对值大于3的Z-score为异常值 outliers np.abs(z_scores) 3 data_clean data.copy() data_clean[outliers] np.nan4. 高级应用与性能优化当数据量很大或需要实时计算时性能成为关键考虑因素。以下是几种优化策略4.1 增量计算移动平均可以通过增量方式高效计算避免重复运算def incremental_moving_average(series, window): result np.empty(len(series)) cumsum series.cumsum() result[window-1:] (cumsum[window-1:] - cumsum[:-window1]) / window result[:window-1] np.nan return result4.2 并行计算对于超大数据集可以使用Dask进行并行计算import dask.array as da dask_data da.from_array(data.values, chunks1000) dask_sma dask_data.map_overlap( lambda x: np.convolve(x, np.ones(5)/5, modevalid), depth4, boundarynone )4.3 结合其他预测方法移动平均可以与其他方法结合提升预测效果from statsmodels.tsa.holtwinters import SimpleExpSmoothing # 先用移动平均平滑数据 smoothed data.rolling(window3).mean().dropna() # 再用指数平滑进行预测 model SimpleExpSmoothing(smoothed) fit model.fit() forecast fit.forecast(3)5. 实战案例销售预测系统让我们把这些技术整合到一个实际的销售预测案例中。假设我们有一家电商过去两年的日销售数据需要预测未来一周的销量。# 加载数据 sales pd.read_csv(daily_sales.csv, parse_dates[date], index_coldate) # 数据预处理 sales sales.resample(D).mean() # 确保每日数据 sales sales.interpolate() # 填充缺失值 # 计算多种移动平均 sales[SMA_7] sales[volume].rolling(7).mean() sales[WMA_7] sales[volume].rolling(7).apply( lambda x: np.dot(x, [0.1, 0.15, 0.2, 0.25, 0.15, 0.1, 0.05]) ) # 评估模型性能 def mape(y_true, y_pred): mask y_true ! 0 return np.mean(np.abs((y_true[mask] - y_pred[mask]) / y_true[mask])) * 100 sma_error mape(sales[volume][7:], sales[SMA_7][7:-1]) wma_error mape(sales[volume][7:], sales[WMA_7][7:-1]) print(fSMA 7天误差率: {sma_error:.2f}%) print(fWMA 7天误差率: {wma_error:.2f}%) # 生成预测 last_window sales[volume][-7:].values next_week_sma np.mean(last_window) next_week_wma np.dot(last_window, [0.1, 0.15, 0.2, 0.25, 0.15, 0.1, 0.05])在这个案例中我们不仅实现了基本的移动平均计算还加入了误差评估和实际预测生成。WMA由于给了近期数据更高权重在这个案例中表现优于SMA。