Question

我的老师给了我这段代码的提纲，但是我不断得到ModuleNotFoundError: No module named 'error'。我知道我需要一个名为error的模块，但是当我找不到名为error的模块的代码时。

我正在尝试解决这个问题：

Solve the tridiagonal equations Ax = b by Doolittle’s decomposition method, where:

A = [6 2 0 0 0
    −1 7 2 0 0
     0 −2 8 2 0    
     0 0 3 7 −2    
     0 0 0 3 5]

b = [2    
    −3    
     4    
    −3    
     1].

这是我使用的代码：

from numpy import argmax, dot, zeros, array, asarray, tril, triu

def swapRows(v,i,j):
    if len(v.shape) == 1: v[i],v[j] = v[j],v[i]
    else:
        temp = v[i].copy()
        v[i] = v[j]
        v[j] = temp
    def swapCols(v,i,j):
    temp = v[:,j].copy()
    v[:,j] = v[:,i]
    v[:,i] = temp
     import error

def LUdecomp(a,tol=1.0e-9):
    n = len(a)
    seq = array(range(n))
       # Set up scale factors
    s = zeros(n)
    for i in range(n):
        s[i] = max(abs(a[i,:]))        
    
    for k in range(0,n-1):
        
      # Row interchange, if needed
        p = argmax(abs(a[k:n,k])/s[k:n]) + k
        if abs(a[p,k]) <  tol: error.err('Matrix is singular')
        if p != k:
            swapRows(s,k,p)
            swapRows(a,k,p)
            swapRows(seq,k,p)
            
      # Elimination            
        for i in range(k+1,n):
            if a[i,k] != 0.0:
                lam = a[i,k]/a[k,k]
                a[i,k+1:n] = a[i,k+1:n] - lam*a[k,k+1:n]
                a[i,k] = lam
    return a,seq

def LUsolve(a,b,seq):
    n = len(a)
       # Rearrange constant vector; store it in [x]
    x = b.copy()
    for i in range(n):
        x[i] = b[seq[i]]
           # Solution
    for k in range(1,n):
        x[k] = x[k] - dot(a[k,0:k],x[0:k])
    x[n-1] = x[n-1]/a[n-1,n-1]    
    for k in range(n-2,-1,-1):
       x[k] = (x[k] - dot(a[k,k+1:n],x[k+1:n]))/a[k,k]
    return x

A = asarray( [ [ 6, 2, 0, 0, 0 ], 
               [ -1, 7, 2, 0, 0 ], 
               [ 0, -2, 8, 2, 0 ], 
               [ 0, 0, 3, 7, -2 ], 
               [ 0, 0, 0, 3, 5 ] ], dtype=float ) A_orig = A.copy() b = asarray( [ 2, -3, 4, -3, 1 ], dtype=float )  b_orig = b.copy() 

A,seq = LUdecomp(A) # A is overwritten as L\U L = tril( A, -1 ) # extract L  for ii in range(L.shape[0]): L[ii,ii] = 1.0 # add in 1's on the diagonal  U = triu( A,  0 ) # extract U  print ("L = ") print (L) print ("U = ") print (U)  if False:
    print ("A[seq,:]= ")
    print (A_orig[seq,:])
    print ("LU= ")
    print (dot(L,U)) 

x = LUsolve(A,b,seq)  print ("Solution= ", x)

Answer 1

如果您打算在某个时候抛出错误，那么无需使用import error语句就可以实现。引发异常可能是一种解决方案。请参阅errors and exceptions上的文档。

您可以删除import error并进行编辑

if abs(a[p,k]) <  tol: error.err('Matrix is singular')

在LUdecomp()中如下：

if abs(a[p,k]) <  tol: 
    raise Exception('Matrix is singular')

我不断收到“ ModuleNotFoundError：没有名为“错误”的模块”

1 个答案: