我要做一个函数,用高斯-雅可比方法来求解一个增广矩阵,但是我不明白为什么我的解总是[0,0,0]。以下是我所拥有的:
from copy import deepcopy
# Define function
def GaussJacobi(MyA, MyX, xtol=1e-6, maxiter=50):
# MyA=Aaug=augmented matrix
# MyX=x=vector(array) containing inital guess
# xtol=exit condition, xnew-xold is less than xtol
# maxiter=max number of iterations allowed before exiting
#b=equation answers from augmented matrix
n=len(MyA)
x=deepcopy(MyX)
sum=0
b = [0] * len(MyA)
lastcol = len(MyA[0]) - 1
for i in range(len(b)):
b[i] = MyA[i][lastcol]
for iter in range(maxiter):
for row in range(n-3):
for col in range(n-3):
sum=b[col]-MyA[row][col]*x[col]
x[row]=sum/MyA[row][col]
return x
MyA = [[4, -1, -1, 3],
[-2, 6, 1, 9],
[-1, 1, 7, -6]]
MyX = [0, 0, 0]
print(GaussJacobi(MyA, MyX, xtol=1e-6, maxiter=50))
首先,您不应该像以前那样使用整数除法 i、 在这种情况下,
3/4 = 0
而不是你应该做的是使用这不是你的代码有问题。你应该试着调试语句,把print语句放在适当的地方,比如设置
x
的地方。这让我看到,你甚至没有进入这一点,因为n = len(myA) = 3
所以n - 3 = 0
所以内部循环被弄乱了。在还有很多其他的问题,因为你没有真正总结你所犯下的术语,只是重新设置了总数-而没有重做整个代码(这是我想要做的!)我不能确定为什么是},非零结果也很可能是错误的。在
0
,但我可以告诉你即使不是{非numpy部分可能是您试图遵循this wiki-您应该复制这个并使用点积的公式。在
相关问题 更多 >
编程相关推荐