拉格朗日插值法-python实现

下面即为拉格朗日插值法的一个实例
我将文件上传到,我的资源当中了,可以直接下载,运行

import pandas as pd
from scipy.interpolate import lagrange #导入拉格朗日插值函数##
#缺失值处理——拉格朗日插值法
inputfile = 'd:/data/catering_sale.xls' #销量数据路径
outputfile = 'd:/data/sales.xls' #输出数据路径 
data = pd.read_excel(inputfile) #读入数据
data[u'销量'][(data[u'销量'] < 400) | (data[u'销量'] > 5000)] = None #过滤异常值,将其变为空值 #自定义列向量插值函数#s为列向量,n为被插值的位置,k为取前后的数据个数,默认为5
def ployinterp_column(s, n, k=5):  
    y = s[list(range(n-k, n)) + list(range(n+1, n+1+k))] #取数  
    y = y[y.notnull()] #剔除空值  
    return lagrange(y.index, list(y))(n) #插值并返回插值结果 #逐个元素判断是否需要插
for i in data.columns:  
        for j in range(len(data)):    
             if (data[i].isnull())[j]: #如果为空即插值。      
data[i][j] = ployinterp_column(data[i], j) 
data.to_excel(outputfile) #输出结果,写入文件 
原文地址:https://www.cnblogs.com/gaoxing2580/p/13193451.html