Julia -如何有效地将矩阵的对角线归零？

# 创建一个示例矩阵
matrix = [1 2 3; 4 5 6; 7 8 9]
println("原始矩阵: ", matrix)

# 获取矩阵的大小
n = size(matrix, 1)

# 将对角线元素设置为0
for i in 1:n
    matrix[i,i] = 0
end

println("对角线归零后的矩阵: ", matrix)

# 创建一个非方阵示例
matrix_non_square = [1 2 3; 4 5 6]
println("原始非方阵: ", matrix_non_square)

# 获取矩阵的行数和列数
rows, cols = size(matrix_non_square)

# 将主对角线元素设置为0
for i in 1:min(rows, cols)
    matrix_non_square[i,i] = 0
end

println("主对角线归零后的非方阵: ", matrix_non_square)