在不使用NumPy的情况下修改Python方阵对角线元素

首先咱们拆解下你遇到的问题：你的函数返回None且矩阵没被修改，核心是两个小疏漏导致的：

1. 你只执行了加法运算，但没有把计算结果赋值回矩阵的对角线元素，相当于白做了计算；
2. 函数没有写return语句，Python里无return的函数默认返回None，所以你打印的是这个默认值。

先回顾下你已有的数据和初始矩阵：

values = [2.1254,0.0125,1.3542,0.2543]
residuals = [0.0251,0.0005,0.1223,0.0043]
matrix = [[i] * len(values) for i in values]

方案1：直接修改原矩阵并返回 #

这个方案会直接修改传入的矩阵，然后返回修改后的结果，代码如下：

def fc(x):
    for i in range(len(values)):
        # 关键：用+=把加法结果赋值回对角线位置，直接修改原元素
        x[i][i] += residuals[i]
    # 必须返回修改后的矩阵
    return x

调用方式：

result_matrix = fc(matrix)
print(result_matrix)

运行后就能得到你期望的矩阵：

[[2.1505, 2.1254, 2.1254, 2.1254],
 [0.0125, 0.013, 0.0125, 0.0125],
 [1.3542, 1.3542, 1.4765, 1.3542],
 [0.2543, 0.2543, 0.2543, 0.2586]]

方案2：创建新矩阵（不修改原数据，更安全） #

如果你不想改动原始的matrix（避免后续代码用到原矩阵时出错），可以先复制原矩阵，再修改新矩阵的对角线：

def fc(x):
    # 复制原矩阵的每一行，生成独立的新矩阵
    new_matrix = [row.copy() for row in x]
    for i in range(len(values)):
        new_matrix[i][i] += residuals[i]
    return new_matrix

调用后原matrix依然保持初始状态，result_matrix是修改后的目标矩阵。

原函数失效的原因复盘 #

再明确下原函数的问题：

• x[i][i] + residuals[i]只是生成了一个临时计算值，没有把这个值存回矩阵，所以矩阵完全没变化；
• 函数没有return x语句，执行完循环后直接结束，返回Python默认的None，你打印的就是这个默认返回值。

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