r - 使用 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包中定义。