Python 中的数组赋值是地址引用,在一些迭代问题的求解中需要用到numpy中的copy函数来完成。
1、 错误的版本
import mumpy as np
# 初值
X = X0
eps = 1e-3
#更新
while True:
X = X0 - np.dot(np.linalg.inv(D(X0)),f(X0))
if np.max(abs(X-X0))<eps:
break
else:
X0 = X #1
这样最多只能更新一次,因为在#1处X0将引用X的地址,所以无论之后X再怎么更新,X0将恒等于X!
2、正确的版本
只需要将#1处改为X的一个copy即可
import numpy as np
# 初值
X = X0
eps = 1e-3
#更新
while True:
X = X0 - np.dot(np.linalg.inv(D(X0)),f(X0))
if np.max(abs(X-X0))<eps:
break
else:
X0 = X.copy() #1
因为一个小问题扯了一晚上,真的是小白啊!