使用R中的滤波器函数的EMA计算

Question

我想用矢量化符号

重现此代码

getEMA2<-function(x,win){
k<-2/(win+1)
v<-vector()
for (i in 1:length(x)){
    if (i==1){
        v[i]<-x[i]
    }
    else{
        v[i]<-k*x[i]+(1-k)*v[i-1]
    }
}
return (v)
}
testOutput<-getEMA2(rnorm(100,0,1),5)

我尝试过使用过滤器功能，但似乎没有递归/卷积方法可以实现这个

感谢您的回复，

Answer 1

由于filter计算

y[n] = x[n] + alpha * y[n-1]

您需要重新调整结果。

f <- function(x,win) {
  alpha <- 2/(win+1)
  filter(x, 1-alpha, method="recursive", side=1, init=x[1]/alpha)*alpha
}
x <- 1:10
k <- 3
getEMA2(x,k)
f(x,k) # identical

大部分过滤器已在TTR包中定义。