从已知矩阵构建块对角矩阵

时间:2017-12-23 18:52:20

标签: r matrix diagonal

我想通过将B放在A的对角线位置,从已知矩阵(B)构建块对角矩阵(A)。

让我们说B:

> matrix(c(1,3,4,5),nrow=2)
      [,1] [,2]
[1,]    1    4
[2,]    3    5 

我正在寻找这样的函数:function(B,3)(3只是一个随机数),它返回矩阵A,如下所示:

[1,] 1 4 . . . .
[2,] 3 5 . . . .
[3,] . . 1 4 . .
[4,] . . 3 5 . .
[5,] . . . . 1 4
[6,] . . . . 3 5

非常感谢任何帮助

2 个答案:

答案 0 :(得分:6)

我们可以使用bdiag

library(Matrix)
bdiag(replicate(3, B, simplify = FALSE))
#6 x 6 sparse Matrix of class "dgCMatrix"

#[1,] 1 4 . . . .
#[2,] 3 5 . . . .
#[3,] . . 1 4 . .
#[4,] . . 3 5 . .
#[5,] . . . . 1 4
#[6,] . . . . 3 5

我们可以包装一个函数

fdiag <- function(mat, n) {
      bdiag(replicate(n, mat, simplify = FALSE))
}

fdiag(B, 3) 

答案 1 :(得分:5)

所需矩阵是单位矩阵和B的kronecker积。没有包使用。

B <- matrix(c(1,3,4,5),nrow=2)
diag(3) %x% B

,并提供:

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