Python實現牛頓插值法(差商表)

def func(x,y,X,infor=True):
    list2=[y[0]]       #  差商表的對角線的第一個元素始終是y0
    count=1
    while(True):
        if len(y)>1:
            list=[]                 # 空列表用來保存,每次計算後差商表的行
            for i in range(len(y)-1):
                n=x[i+count]-x[i]
                m=y[i+1]-y[i]
                l=m/n
                list.append(l)
            list2.append(list[0])        # list2用來記錄差商表的對角線元素,每計算一次,取行的第一個元素
            count += 1
            y = list
        else:
            break
    if infor:                           # 判斷是否要繼續計算,結果
        W=0
        for i in range(len(list2)):
            if i==0:
                w=list2[i]
            else:
                w = list2[i]
                for j in range(i):
                    w*=(X-x[j])
            W+=w
        print('牛頓插值:', W)
    return '牛頓差商表對角線列:%s' %list2

ret=func([0.32, 0.34, 0.36],[0.314567, 0.333487, 0.352274],'',infor=False)
print(ret)
ret=func([0.32, 0.34, 0.36],[0.314567, 0.333487, 0.352274],0.3367)
print(ret)

運行結果: