r - 提取大矩阵的非对角线切片

我有一个很大的nxn矩阵，想取不同大小的非对角线切片。例如：

1 2 3 4 5 6
1 2 3 4 5 6
1 2 3 4 5 6
1 2 3 4 5 6
1 2 3 4 5 6
1 2 3 4 5 6

我想要一个R函数，当给定矩阵和“对角切片的宽度”时，它将返回仅包含这些值的nxn矩阵。因此，对于上面的矩阵，例如3，我将得到：

1 x x x x x
1 2 x x x x
1 2 3 x x x
x 2 3 4 x x
x x 3 4 5 x
x x x 4 5 6

目前，我正在使用（原谅我）一个for循环，它非常慢：

getDiags<-function(ndiags, cormat){
  resmat=matrix(ncol=ncol(cormat),nrow=nrow(cormat))
  dimnames(resmat)<-dimnames(cormat)
  for(j in 1:ndiags){
    resmat[row(resmat) == col(resmat) + j] <-
      cormat[row(cormat) == col(cormat) + j]
  }
  return(resmat)
}

我意识到这是解决此问题的一种非常“ un-R”的方式。有没有更好的方法，可能使用diag或lower.tri？

最佳答案

size <- 6
mat <- matrix(seq_len(size ^ 2), ncol = size)


low <- 0
high <- 3

delta <- rep(seq_len(ncol(mat)), nrow(mat)) -
    rep(seq_len(nrow(mat)), each = ncol(mat))
#or Ben Bolker's better alternative
delta <- row(mat) - col(mat)
mat[delta < low | delta > high] <- NA
mat

这可以在我的机器上使用5000 x 5000矩阵