R语言 diag函数的用法

R语言 diag函数的用法

我们今天写一下关于R语言中diag函数的用法：
（1）diag函数可以用来提取矩阵的对角元素，并将它保存成向量的格式。值得注意的是,如果所要提取的矩阵不是方阵的话，diag函数仍然可以调用。

>   a <- matrix(c(1,2,3,4),nrow=2)
>   diag(a)
[1] 1,4
>   b <- matrix(c(1,2,3,4,5,6),nrow=2)
[1] 1,4

（2）diag函数可以用来生成给定对角元素的对角阵。

>   a <- c(1,4)
>   diag(a)
     [,1] [,2]
[1,]   1   0
[2,]   0   4

（3）如果在diag( )函数的括号内添加的元素是一个标量，那么可以生成指定行列的单位阵。

>   diag(2)
     [,1] [,2]
[1,]   1   0
[2,]   0   1

注：未经本文作者同意，严禁转载本篇文章。