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 (длина (х), размер, замена проб): неправильное количество вероятностей

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