r - 跨列随机分配总和

Question

假设我有一个长度为 n 的向量.比方说 n=3所以向量是 v=c(3,5,4) .

我有一个零矩阵 n行和 m列。比方说 m=5

mymatrix <- matrix(rep(0, 3*5), nrow=3)

我想要的是随机分布 v 的值跨列，针对每一行。所以在这个例子中，第一行总和为 3，第二行总和为 5，第三行总和为 4。这将是一种可能的随机分配:

0   1   2   0   0
3   1   0   0   1
0   2   2   0   0

行的总和为 3,5,4，即 v 的值.

我怎样才能做到这一点？我的想法是开始

sapply(v, function(i){sample(1:m, i, replace=TRUE)})

然后从那里开始，但这给了我一个列表，因为每个结果的长度不同，我不知道如何从那里开始。

编辑:意图不是负数，所以 0 9 1 1 -8总和为 3 将不是有效行。

Answer 1

使用 purrr :

library(tibble)
library(dplyr)
library(purrr)

n <- 3
m <- 5
v <- c(3,5,4)
y <- sapply(v, function(i){sample(1:m, i, replace=TRUE)})

do_it <- function(x) {
  tmp <- tibble(
    index = x,
    cnt = 1
  ) %>% 
    group_by(index) %>% 
    summarise(cnt = sum(cnt))
  out <- rep(0, m)
  out[tmp$index] <- tmp$cnt
  return(out)
}

y %>%
  map(~do_it(.x)) %>% 
  unlist() %>% 
  matrix(nrow = 3, byrow = TRUE)