Question

我试图用1和0来在R中构建矩阵，但是只有满足行和列条件时，才应生成矩阵。例如：

我想用rowsums = c(3, 2, 4, 2)和colsums = c(3, 2, 4, 2)创建一个矩阵。

Answer 1

也许有一个更优雅的解决方案，但这可以正常工作，并且运行时间不长，并且不需要任何软件包：

repeat {
  repeat {
    repeat {
      repeat {
        repeat {
          x <- sample(0:1, 4, replace = T)
          if(sum(x) == 3) {
            break
          }
        }
        repeat {
          y <- sample(0:1, 4, replace = T)
          if(sum(y) == 2) {
            break
          }
        }
        repeat {
          z <- sample(0:1, 4, replace = T)
          if(sum(z) == 4) {
            break
          }
        }
        repeat {
          u <- sample(0:1, 4, replace = T)
          if(sum(u) == 2) {
            break
          }
        }
        if(sum(x[1], y[1], z[1], u[1]) == 3) {
          break
        }
      }
      if(sum(x[2], y[2], z[2], u[2]) == 2) {
        break
      }
    }
    if(sum(x[3], y[3], z[3], u[3]) == 4) {
      break
    }
  }
  if(sum(x[4], y[4], z[4], u[4]) == 2) {
    print(matrix(c(x, y, z, u), 4, 4, byrow = T))
    break
  }
}

Answer 2

将其视为lp问题的一个问题，而phiver也在Stephane Laurent在评论中提供的链接中提到了该问题：

library(lpSolve)
library(data.table)

nr <- 4
nc <- 4
ne <- nr * nr

v <- vector("integer", ne)
colvec <- replace(v, seq_along(v) <= nr, 1L)
rowvec <- replace(v, seq_along(v) %% nc == 1, 1L)

colconstr <- c(3, 2, 4, 2)
rowconstr <- c(3, 2, 4, 2)

const.mat <- do.call(rbind, c(
    data.table::shift(colvec, seq(0L, ne - nc, nc), fill=0L),
    data.table::shift(rowvec, 0L:(nr-1L), fill=0L)))
const.rhs <- c(colconstr, rowconstr)

s <- lpSolve::lp("min", runif(ne), const.mat, 
    rep("==", nrow(const.mat)), const.rhs, all.bin=TRUE)$solution
matrix(s, nrow=nr)

示例输出：

     [,1] [,2] [,3] [,4]
[1,]    1    0    1    1
[2,]    1    0    1    0
[3,]    1    1    1    1
[4,]    0    1    1    0

基于行和列总和的条件随机矩阵为0和1

2 个答案: