Python实现拉格朗日插值法填补缺失值

朝雾初升、

已于 2023-02-23 13:10:28 修改

阅读量3.8k

点赞数 6

文章标签： python 数据分析机器学习

于 2021-08-18 17:27:54 首次发布

本文链接：https://blog.csdn.net/qq_42871783/article/details/119785149

版权

# s为列向量，n为被插值的位置，k为取前后的数据个数，默认为5
from scipy.interpolate import lagrange  # 导入拉格朗日插值函数
def ployinterp_column(s, n, k=5): 
    y = s.iloc[list(range(n-k, n)) + list(range(n+1, n+1+k))]  # 取数
    y = y[y.notnull()]  # 剔除空值
    return lagrange(y.index, list(y))(n)  # 插值并返回插值结果

def data_lagrange(data):  
    for i in data.columns:   #获取data的列名
        for j in range(len(data)):
            if (data[i].isnull())[j]:  # 判断data的i列第j个位置的数据是否为空，如果为空即插值
                data.loc[j,i] = ployinterp_column(data[i], j)
    return data```
#填补data中的缺失值
data_lagrange(data)