計算方法——ラグランジュ補間多項式

計算方法——ラグランジュ補間多項式
【問題の説明】[0.0,1.2]内の関数y=f(x)=cos(x)を考慮する.ラグランジュ補間多項式は、複数(2,3,4等)のノードを用いて構築される.
【入力形式】画面に区間[0.0,1.2]内の1つの値x*を順次入力し、補間多項式を構築してそのP(x*)値と、複数のノードのx座標を求める.
【出力形式】補間多項式係数行列、ラグランジュ係数多項式行列、P(x*)値(小数点以下6桁有効数字保持)を出力します.
【サンプル1入力】
0.3
0 0.6 1.2
【サンプル1出力】
[-0.40043538 -0.05084608 1. ]
[[ 1.38888889 -2.5 1. ]
[-2.77777778 3.33333333 -0. ]
[ 1.38888889 -0.83333333 0. ]]
P2(0.3)=0.948707
【採点基準】入力による出力が正確

import numpy as np
import math
xx=input()
xx=float(xx)
Enter = input()
Enter = Enter.split(" ")
n= len(Enter)
X=np.zeros((1,n))
Y=np.zeros((1,n))
A=np.zeros((n,n))
X[0]=Enter
fxx=[]
fxxx=np.poly1d([0,0])

for i in range(0,n):
    Y[0][i]=math.cos(X[0][i])

for i in range(0,n):
    fxx.append(np.poly1d([0,1]))

for i in range(0,n):
    v=np.poly1d([0,1])
    for j in range(0,n):
        if j!=i :
            fxx[i]=np.polymul(fxx[i],np.poly1d([1,-X[0][j]]))/(X[0][i]-X[0][j])
            v=np.polymul(v,np.poly1d([1,-X[0][j]]))/(X[0][i]-X[0][j])
    A[i]=v.c

for i in range(0,n):
    fxxx=fxx[i]*Y[0][i]+fxxx

print(fxxx.c)
print(A)
print('P{}({})={:.6f}'.format(n-1,xx,fxxx(xx)))


'''''
0.3
0 0.6 1.2
'''''