在R中创建身份矩阵

Question

我想创建不同大小的标识矩阵，我可以在更小的范围内这样做：

> x <- matrix(cbind(c(1,0), c(0,1)), 2)
> x
     [,1] [,2]
[1,]    1    0
[2,]    0    1

就像这样：

> y <- matrix(cbind(c(1,0,0), c(0,1,0), c(0,0,1)), 3)
> y
     [,1] [,2] [,3]
[1,]    1    0    0
[2,]    0    1    0
[3,]    0    0    1

然而，随着身份矩阵规模的扩大，这似乎会变得越来越乏味。

是否有更简单的方法来创建 n - 值标识矩阵？

Answer 1

指定diag时nrow的一两（2）次使用，或当x是长度为1的向量时，您会得到一个单位矩阵

diag(5)
diag(nrow = 5)

或者您可以创建一个0的矩阵并填充对角线：

mat <- matrix(0, 5, 5)
diag(mat) <- 1

## or shorter:
`diag<-`(matrix(0, 5, 5), 1)

所有这些都让我：

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