R：如何編寫給定概率采樣的函數

Question

我寫了一個函數

p = c(0.4, 0.6)
myfun = function(p){
    sample(1:2, 1, replace = TRUE, prob = p)
}

我想重復5次。

sapply(1:5, myfun)

但這給我一個錯誤

sample.int（長度（x），大小，替換，概率）中的錯誤：概率數不正確

Answer 1

您可以使用replicate編寫函數。

myfun <- function(x, p, n, replace = TRUE) {
    m <- replicate(n, sample(x, replace = replace, prob = p))
    if(n == 1) c(m) else m
}
myfun(2, c(0.4, 0.6), 5)
#      [,1] [,2] [,3] [,4] [,5]
# [1,]    1    1    2    2    2
# [2,]    2    2    2    1    2
myfun(2, c(0.4, 0.6), 1)
# [1] 2 1

Answer 2

您可以使用sample函數對5個值進行采樣。

p <- c(0.4, 0.6)
sample(1:2, 5, replace = TRUE, prob = p)
# [1] 2 2 2 2 1