移动平均滤波-python实现

#  读取数据
import pandas as pd
import numpy as np

file_path = 'dst_ess.csv'
df = pd.read_csv(file_path, index_col = False, engine = 'python').dropna()

# n_iter =  508 #10000
n_iter = df['风速'].__len__()
array = np.array(df['风速'][0:n_iter]) # observations (normal about x, sigma=0.1)

################ 移动平均滤波 #############################

class MovingAverageFilter:
	"""Simple moving average filter"""

	@property
	def avg(self):
		"""Returns current moving average value"""
		return self.__avg

	def __init__(self, n = 20, initial_value = 0):
		"""Inits filter with window size n and initial value"""
		self.__n = n
		self.__buffer = [initial_value/n]*n
		self.__avg = initial_value
		self.__p = 0

	def __call__(self, value):
		"""Consumes next input value"""
		self.__avg -= self.__buffer[self.__p]
		self.__buffer[self.__p] = value/self.__n
		self.__avg += self.__buffer[self.__p]
		self.__p = (self.__p  + 1) % self.__n
		return self.__avg


moving_average_filter = MovingAverageFilter(n= 6, initial_value = array[0])

array_dest = []
for num in array:
	moving_average_filter.__call__(num)
	# ret = moving_average_filter.__call__(num)
	ret = moving_average_filter.avg
	array_dest.append ( ret )

#  显示数据
import matplotlib.pyplot as plt
import numpy as np
plt.rcParams['axes.unicode_minus']=False #用来正常显示负号

plt.figure()
plt.plot(array)
plt.plot(array_dest)
plt.show()