R - 从矩阵中采样"row-wise"

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中的概率从每行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

您可以使用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

Answer 2

一个带有mapply和asplit的选项

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