r - 从数组中选择一个随机数并返回它的索引

Question

我有一系列概率

A = c(0, 0, 1/3, 0, 1/3, 1/3)

我使用的随机数公式是：

sample(A, 1, replace=TRUE)

它给了我一种可能性。我要找的是对应概率的索引号。由于概率可以重复，我将如何获得选择的特定索引号？此外，它不应该能够选择 0 概率。

谢谢你。

Answer 1

制作一个基本上这样做的函数：

y <- 0 
while(y == 0) { 
  x <- sample(1:length(A), 1) 
  y <- A[x]
}
print(x)

replace = TRUE 在这种情况下是多余的，因为您一次只采集一个样本。