如何用4000x1列向量生成对角矩阵（其余元素为0）

嘿，这个需求在R里其实超容易解决，用内置的diag()函数就能一步到位！

核心方法 #

R的diag()函数专门用来创建或提取对角矩阵，当你传入一个向量（或者可以转换为向量的列矩阵）时，它会自动生成一个对角矩阵——对角线元素就是你传入的向量元素，其余位置全为0。

对应你的示例操作 #

首先你定义的列向量：

vector <- matrix(c(1:5), ncol=1, nrow=5)

只需要一行代码就能生成目标矩阵：

# 生成对角矩阵
diag_matrix <- diag(as.vector(vector))

（注：as.vector()是把列矩阵转换成一维向量，其实直接写diag(vector)也能得到同样结果，因为diag()会自动提取矩阵的元素作为向量，加as.vector()只是让逻辑更清晰~）

运行后得到的diag_matrix就是你想要的格式：

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

针对你4000x1的列向量 #

用法完全一致，不管你的列向量长度是多少，diag()都能轻松处理：

# 假设你的4000x1列向量名为my_vector
large_diag_matrix <- diag(as.vector(my_vector))

生成的large_diag_matrix会是一个4000×4000的矩阵，对角线就是你输入的列向量元素，其余位置全为0，而且这个内置函数效率很高，处理大矩阵也不会有性能问题~

内容的提问来源于stack exchange，提问作者nicolas garzon