Julia中矩阵对角线赋值报错，如何解决？

你遇到的问题本质是误用了Julia的语法——你写的diag(aa) = ones(3)并不是给矩阵对角线赋值的正确方式，而是在尝试定义/扩展LinAlg.diag函数，这就是为什么会抛出那个关于“必须显式导入才能扩展”的错误。

在Julia里，diag()函数默认是用来提取矩阵对角线元素的，不能直接通过它来修改原矩阵的元素。下面给你几种正确的解决方法：

方法1：用diagind()获取对角线索引赋值（最简洁） #

diagind(aa)会返回矩阵aa所有对角线元素的索引，直接对这些位置赋值即可：

# 先生成随机矩阵
aa = rand(3,3)
# 把对角线设为1，用广播赋值更简洁
aa[diagind(aa)] .= 1

如果需要赋值为自定义的数组（比如[1,2,3]），也可以直接写：

aa[diagind(aa)] = [1,2,3]

方法2：循环遍历对角线位置（最直观） #

如果你觉得索引方式不好理解，直接通过循环访问每个对角线元素的位置(i,i)来赋值：

aa = rand(3,3)
for i in 1:size(aa, 1)
    aa[i, i] = 1
end

这种方法逻辑清晰，适合新手理解矩阵对角线的位置规律。

方法3：生成新矩阵（不修改原矩阵） #

如果你不想改动原矩阵，而是要得到一个对角线为1、其他元素和原矩阵一致的新矩阵，可以借助LinearAlgebra包的功能：

using LinearAlgebra

aa = rand(3,3)
# 减去原对角线矩阵，再加单位矩阵I，得到对角线为1的新矩阵
new_aa = aa - Diagonal(aa) + I

补充说明 #

记得不要混淆“函数调用”和“赋值操作”：diag(aa)是调用函数提取元素，返回的是一个新数组，直接给它赋值毫无意义（甚至会被当成函数定义），必须直接操作原矩阵的索引位置才能修改它的元素。

内容的提问来源于stack exchange，提问作者Xia.Song