Python移动平均实战教程详解

小伙伴们有没有觉得学习文章很有意思？有意思就对了！今天就给大家带来《Python数据平滑技巧：移动平均实战教程》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

移动平均是一种常用的数据平滑方法，通过计算连续数据点的平均值来减少噪声并突出趋势。Python中可用NumPy和Pandas实现，如使用np.convolve或pd.Series.rolling().mean()进行简单移动平均（SMA），以及pd.Series.ewm().mean()进行指数移动平均（EMA）。窗口大小的选择需根据数据周期性、实际效果及领域知识调整，过小则平滑不足，过大则可能丢失特征。移动平均的变种包括：1. SMA所有点权重相同；2. 加权移动平均（WMA）为不同点分配不同权重；3. EMA权重呈指数衰减，更关注近期数据。其他平滑方法还有：Savitzky-Golay滤波器、小波变换、卡尔曼滤波和LOESS，各自适用于不同场景，如保留局部特征、系统状态估计或未知分布数据。

数据平滑，简单来说，就是减少数据中的噪声，让趋势更明显。Python有很多库可以做到这一点，移动平均是其中一种比较简单且常用的方法。

移动平均处理，就是用一系列连续数据点的平均值来代替原始数据点。这能有效过滤掉短期波动，突出长期趋势。

解决方案：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

def moving_average(data, window_size):
  """
  计算移动平均。

  Args:
    data: 原始数据，列表或NumPy数组。
    window_size: 窗口大小，即用于计算平均值的连续数据点数量。

  Returns:
    移动平均后的数据，列表。如果窗口大小大于数据长度，返回空列表。
  """
  if window_size > len(data):
    return [] # 处理窗口过大的情况

  # 使用NumPy进行优化
  data_as_array = np.asarray(data)
  window = np.ones(window_size) / window_size
  smoothed_data = np.convolve(data_as_array, window, mode='valid')
  return smoothed_data.tolist()


# 示例数据
data = [1, 2, 3, 4, 5, 4, 3, 2, 1, 2, 3, 4, 5]

# 设置窗口大小
window_size = 3

# 计算移动平均
smoothed_data = moving_average(data, window_size)

# 打印结果
print("原始数据:", data)
print("移动平均后的数据:", smoothed_data)

# 可视化
plt.plot(data, label='Original Data')
plt.plot(smoothed_data, label='Moving Average (Window Size = {})'.format(window_size))
plt.legend()
plt.xlabel('Data Point Index')
plt.ylabel('Value')
plt.title('Moving Average Smoothing')
plt.show()


# 使用Pandas实现更灵活的移动平均

# 创建Pandas Series
data_series = pd.Series(data)

# 计算简单移动平均 (SMA)
window_size_pandas = 3
sma = data_series.rolling(window=window_size_pandas).mean()
sma = sma.dropna() # 删除NaN值，因为前几个数据点无法计算均值

print("\nPandas SMA:", sma.tolist())

# 计算指数移动平均 (EMA)
ema = data_series.ewm(span=window_size_pandas, adjust=False).mean() # adjust=False 更稳定
print("Pandas EMA:", ema.tolist())


# Pandas可视化
plt.figure(figsize=(10, 6))
plt.plot(data, label='Original Data')
plt.plot(sma, label='SMA (Window Size = {})'.format(window_size_pandas))
plt.plot(ema, label='EMA (Span = {})'.format(window_size_pandas))
plt.legend()
plt.xlabel('Data Point Index')
plt.ylabel('Value')
plt.title('Pandas Moving Average Smoothing')
plt.show()

移动平均的窗口大小如何选择？

窗口大小的选择至关重要。太小，平滑效果不明显；太大，可能会过度平滑，丢失重要特征。 一般来说，可以根据数据的周期性来选择窗口大小。例如，如果数据存在明显的季节性，可以尝试将窗口大小设置为季节周期长度。 也可以通过尝试不同的窗口大小，观察平滑效果，然后选择一个合适的。 还可以结合领域知识来判断。

移动平均有哪些变种？它们有什么区别？

移动平均有很多变种，常见的有：

• 简单移动平均 (SMA)：所有数据点的权重相同。上面代码示例已经展示。
• 加权移动平均 (WMA)：数据点的权重不同，通常越靠近当前时间点的数据权重越高。
• 指数移动平均 (EMA)：也是一种加权移动平均，但权重呈指数衰减。EMA 对最近的数据点更敏感，反应速度更快。Pandas的ewm函数可以实现EMA。

SMA简单易懂，但对所有数据点一视同仁，可能无法很好地反映趋势变化。WMA和EMA则更加灵活，可以根据实际需求调整权重，更好地捕捉趋势。EMA的计算效率通常比WMA更高。

除了移动平均，还有哪些其他数据平滑方法？

除了移动平均，还有很多其他数据平滑方法，例如：

• Savitzky-Golay 滤波器：这是一种基于多项式拟合的滤波器，可以在平滑数据的同时，保留数据的局部特征。SciPy库提供了savgol_filter函数来实现Savitzky-Golay滤波。
• 小波变换：小波变换可以将数据分解成不同频率的成分，然后可以对不同频率的成分进行处理，从而实现数据平滑。PyWavelets库提供了小波变换的功能。
• 卡尔曼滤波：卡尔曼滤波是一种递归的滤波算法，可以用于估计系统的状态。卡尔曼滤波需要建立系统的状态空间模型，并假设噪声服从高斯分布。
• LOESS (Locally Estimated Scatterplot Smoothing)：LOESS是一种非参数的回归方法，它通过局部加权回归来平滑数据。 statsmodels库提供了LOESS的实现。

选择哪种平滑方法取决于数据的特点和具体需求。如果数据噪声较大，且对平滑效果要求较高，可以考虑使用Savitzky-Golay滤波器或小波变换。如果需要对系统状态进行估计，可以考虑使用卡尔曼滤波。如果数据分布未知，可以考虑使用LOESS。

今天带大家了解了的相关知识，希望对你有所帮助；关于文章的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~