# 读取数据
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()
移动平均滤波-python实现
最新推荐文章于 2024-08-07 17:58:59 发布