解带状联立方程时的溢出误差

Question

我正在尝试使用计算物理学书中的函数来求解带状联立方程：http://www-personal.umich.edu/~mejn/cp/programs/banded.py 但是，运行该函数会导致以下错误：

OverflowError: cannot convert float infinity to integer in banded line 39: v[m] /= div

矩阵A和w为：

A = 
[[ 3 -1 -1 ...  0  0  0]
 [-1  4 -1 ...  0  0  0]
 [-1 -1  4 ...  0  0  0]
 ...
 [ 0  0  0 ...  4 -1 -1]
 [ 0  0  0 ... -1  4 -1]
 [ 0  0  0 ... -1 -1  3

w = [5 5 0 ... 0 0 0]

x = banded(A,w,2,2)

有人可以帮我弄清楚为什么使用此功能时会出现溢出错误吗？ 谢谢。

Answer 1

我认为您应该将A放入banded.py

中所示的带状矩阵中

(  -   -  A02 A13 A24 ...
(  -  A01 A12 A23 A34 ...
( A00 A11 A22 A33 A44 ...
( A10 A21 A32 A43 A54 ...
( A20 A31 A42 A53 A64 ...

此外，我发现如果矩阵A或向量w中的矩阵元素是整数，则带状函数将无法正常工作，您可以考虑将其初始化为浮点型。