baseline2_ARMA.py 文件源码

def draw_trend(timeSeries, size):
    rol_mean = pd.rolling_mean(timeSeries,size)
    rol_weighted_mean = pd.ewma(timeSeries, span=size)
    # f = plt.figure(facecolor='white')
    # plt.plot(timeSeries,'b',rol_mean,'g',rol_weighted_mean,'r')
    # plt.legend(loc='best')
    # plt.title('Rolling Mean')
    # plt.show()
    return  rol_weighted_mean