如何使`integrate()`接受R函数中的向量?

时间:2017-06-20 17:57:00

标签: r function vectorization integral numerical-integration

我想知道如何将我的功能Bpp设为 接受第一个参数的矢量 t

Bpp = function(t, n1, n2 = NULL){

      N = ifelse(is.null(n2), n1, n1*n2/(n1+n2))
     df = ifelse(is.null(n2), n1 - 1, n1 + n2 - 2)

     H1 = integrate(function(delta)dcauchy(delta, 0, sqrt(2)/2)*dt(t, df, delta*sqrt(N)), -Inf, Inf)[[1]]
     H0 = dt(t, df)
   BF10 = H1/H0
p.value = 2*(1-pt(abs(t), df))

list(BF10 = BF10, p.value = p.value)
}

Bpp(t = -6:6, 20, 20) ## This will give error because `t` is now a vector?

1 个答案:

答案 0 :(得分:2)

看起来我可以在没有测试的情况下给出快速回答。在Bpp

中使用以下内容
# joint density
joint <- function(delta, t) dcauchy(delta, 0, sqrt(2)/2) * dt(t, df, delta*sqrt(N))
# marginal density of `t`
marginal.t <- function (t) integrate(joint, lower = -Inf, upper = Inf, t = t)[[1]]
H1 <- sapply(t, marginal.t)
  

那么,在这里我们也可以使用Vectorize看起来怎么样?

使用原始Bpp

Bpp <- Vectorize(Bpp, vectorize.args = "t")
Bpp(-6:6, 20, 20)