R - 来自 Matrix 的“逐行”采样

Question

假设我有 2x 矩阵：

x <- matrix(rep(1:5, 2), ncol=5, byrow=T)
prob <- matrix(rep(0, each=5, 2), ncol=5, byrow=T)
prob[1, 3] <- 1
prob[2, 4] <- 1

> x
     [,1] [,2] [,3] [,4] [,5]
[1,]    1    2    3    4    5
[2,]    1    2    3    4    5

> prob
     [,1] [,2] [,3] [,4] [,5]
[1,]    0    0    1    0    0
[2,]    0    0    0    1    0

我想使用sample()从矩阵中执行“逐行”采样。 例如， sample将使用prob同一行中的prob从x每一行中选择一个值。 在上面的示例中，我希望sample每次都返回c(3,4)因为第 1 行的概率设置为 3=100%，第 2 行的概率设置为 4=100%。

但是，当我运行sample(x, prob=prob, size=nrow(x)) ，我收到3和4是任何顺序：

> sample(x, prob=prob, size=nrow(x))
[1] 4 3
> sample(x, prob=prob, size=nrow(x))
[1] 3 4

如何对矩阵x执行“逐行”采样？

Answer 1

一种带有mapply和asplit选项

mapply(sample, asplit(x, 1 ), prob = asplit(prob, 1), size = 1)
#[1] 3 4

Answer 2

您可以使用vapply （通过指定您期望的内容更安全）来循环每一行：

x <- matrix(rep(1:5, 2), ncol=5, byrow=T)
prob <- matrix(rep(0, each=5, 2), ncol=5, byrow=T)
prob[1, 3] <- 1
prob[2, 4] <- 1


vapply(seq_len(nrow(x)), function(i) {
  sample(x[i, ], prob = prob[i, ], size = 1)
}, FUN.VALUE = numeric(1)
)
#> [1] 3 4