如何在R脚本的If-Else语句中实现概率性分支逻辑？

直接用R的sample()函数来实现概率选择，这是最简洁且符合R风格的写法：

# 根据w0[i,j]的值确定候选选项和对应概率
if (w0[i, j] == 1) {
  candidates <- c(rL[nNeigh + 1], rL[nNeigh])
  probs <- c(0.7, 0.3)
} else {
  candidates <- c(rD[nNeigh + 1], rL[nNeigh])
  probs <- c(0.7, 0.3)
}
# 按概率随机选择赋值
w1[i, j] <- sample(candidates, size = 1, prob = probs)

也可以把逻辑压缩成更紧凑的形式（可读性略降）：

w1[i, j] <- if (w0[i,j] == 1) {
  sample(c(rL[nNeigh+1], rL[nNeigh]), 1, prob = c(0.7, 0.3))
} else {
  sample(c(rD[nNeigh+1], rL[nNeigh]), 1, prob = c(0.7, 0.3))
}

为什么||不行？ #

||是R里的逻辑或运算符，仅用于判断两个条件是否至少有一个为真，完全不适合实现概率选择逻辑。你需要的是基于概率的随机抽样，而非逻辑判断，因此必须用抽样函数来实现。

补充说明 #

• sample()的prob参数用于指定每个候选值的选中概率，参数值总和必须为1（这里0.7+0.3=1，符合要求）。
• 如果需要对整个矩阵批量处理（而非单个[i,j]元素），可以考虑向量化操作，比如结合ifelse()与sample()的向量化用法，但针对单个元素的场景，上面的代码已足够。

内容的提问来源于stack exchange，提问作者BookishWolf