如何仅用R语言matrix()函数创建对角矩阵？代码问题排查

问题原因 #

你给出的代码中，row和col是单个数值（标量），row == col仅会返回一个TRUE值，因此if语句输出单个1。matrix()函数会将这个单一值重复填充到所有矩阵位置，最终得到全1矩阵，而非预期的单位对角矩阵。

要实现“行号等于列号时填1，否则填0”的逻辑，需要对每个元素的行号和列号做逐元素比较，而非比较整体的行列总数。

正确实现方式 #

以下是几种符合要求的一行代码实现：

方法1：利用outer()逐元素比较

通过outer()生成行号与列号的逐元素比较结果，再转成数值型：

row <- 5
col <- 5
matrix(data = outer(1:row, 1:col, `==`) + 0, nrow = row, ncol = col)

逻辑说明：outer(1:row, 1:col, ==)会生成一个逻辑矩阵，其中行号等于列号的位置为TRUE，其余为FALSE；加0会将逻辑值转为数值（TRUE→1，FALSE→0），再传入matrix()生成最终矩阵。

方法2：替换矩阵对角线元素

通过diag<-函数直接给0矩阵的对角线赋值1：

row <- 5
col <- 5
`diag<-`(matrix(0, nrow = row, ncol = col), 1)

逻辑说明：先生成全0矩阵，再用diag<-语法将对角线元素替换为1，实现单位对角矩阵。

方法3：直接使用R内置的diag()函数

如果无需严格限定用matrix()函数，R内置的diag()是最简洁的方式：

diag(5)

该函数可直接生成指定阶数的单位对角矩阵，效率更高。

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