使用numpy.zeros时遇TypeError：NoneType与float无法相加的问题求助

问题出在np.fill_diagonal的使用方式上：这个函数是原地修改数组，执行后不会返回修改后的矩阵，而是返回None。你把它的返回值赋值给了lamb_diag，直接把原本的全零矩阵覆盖成了None，后续自然无法和sigma执行相加操作。

修正方法有两种：

方法一：保留原地操作，取消赋值 #

lamb_diag = np.zeros((N,N), float)
np.fill_diagonal(lamb_diag, self.lamb)  # 直接调用，不赋值给变量

sigma = lamb_diag + sigma

方法二：直接生成对角矩阵（更简洁） #

用np.diag直接创建对角元素为self.lamb的N阶矩阵，跳过全零矩阵初始化步骤：

lamb_diag = np.diag([self.lamb] * N)

sigma = lamb_diag + sigma

两种方法都能让lamb_diag成为正常的numpy矩阵，顺利执行后续的加法操作。

内容的提问来源于stack exchange，提问作者oéoéoéeuuuh