r - 以概率选择特定数字

Question

如何以特定概率p选择一个数字？

假设我们必须在 {0, 1} 之间进行选择，而 p 代表选择 1 的概率。

因此，当 p=0.8 时，我们选择 1 为 80%，0 为 20%。

R 中有一个简单的解决方案吗？

Answer 1

看看sample功能。

> set.seed(1)
> sample(c(0,1), size=10, replace=TRUE, prob=c(0.2,0.8))
 [1] 1 1 1 0 1 0 0 1 1 1

从帮助文件中您可以阅读:

sample takes a sample of the specified size from the elements of x using either with or without replacement.

并且 sample 中的参数 prob 充当...

A vector of probability weights for obtaining the elements of the vector being sampled.