Question

我找到了一个类似问题的解决方案，当矩阵可能不是正方形时（因为它可能不在我的情况下）Function to split a matrix into sub-matrices in R，它将矩阵分解为一组非重叠的子矩阵。但是，就我而言，我希望矩阵重叠。使用以下矩阵作为示例：

M <- matrix(1:20, 5)

#     [,1] [,2] [,3] [,4]
#[1,]    1    6   11   16
#[2,]    2    7   12   17
#[3,]    3    8   13   18
#[4,]    4    9   14   19
#[5,]    5   10   15   20

如果我选择子矩阵的行数和列数分别为4和4，那么返回的子矩阵应该是每个可能重叠的4X4子矩阵，它完全适合原始矩阵的边界（这是另一个地方，这个问题的答案可能需要与我引用的问题不同）。在我的例子中，只有两个子矩阵应该返回。 M[1:4,1:4]和M[2:5,1:4]。我应该能够选择任意大小的子矩阵。我似乎无法找到任何提取重叠子矩阵的例子，但我可能会过度思考这个。有没有人对最佳方法有任何想法？

Answer 1

也许像这样的策略可以运作

submat <- function(m, nrow, ncol) {
    stopifnot(nrow(m)>=nrow, ncol(m)>=ncol)
    rowstarts<-1:(nrow(m)-nrow+1)
    colstarts<-1:(ncol(m)-ncol+1)
    ss <- function(r, c) {
        m[r:(r+nrow-1), c:(c+ncol-1), drop=FALSE]
    }
    with(expand.grid(r=rowstarts, c=colstarts), mapply(ss, r, c, SIMPLIFY=FALSE))
}

submat(M, 4, 4)

我们确定行和列的可能起始索引的位置，然后我们使用expand.grid()生成这些起始值的所有可能组合，然后我们使用mapply来提取每个可能的子矩阵起始位置。

将矩阵拆分为子矩阵

1 个答案: