R语言中diag函数

R语言中diag函数用于获取矩阵的对角线元素

001、

dat <- matrix(1:9, nrow = 3)     ## 生成矩阵（方阵，行列相等）
dat
diag(dat)                        ## 取对角线元素

002、非方阵情况

dat <- matrix(1:15, nrow = 3)     ## 生成3行5列矩阵
dat
diag(dat)                        ## 取对角线元素

002、生成指定对角线元素的方阵

idx <- c(8, 3, 2, 7)         ## 对角线元素
diag(idx)                    ## 生成矩阵

003、生成指定长度的单位矩阵

diag(5)         ## 生成对角线长度为5的单位矩阵