pandas isnull函数

#-*- coding: utf-8 -*-
#拉格朗日插值
import pandas as pd
from scipy.interpolate import lagrange

inputfile = '../data/missing_data.xls' #输入数据路径,需要使用Excel格式;
outputfile = '../tmp/missing_data_processed.xls' #输出数据路径,需要使用Excel格式

data = pd.read_excel(inputfile) #读入数据

#自定义列向量插值函数
#s为列向量,n为被插值的位置,k为取值前后的数据个数,暂定为5
def ployinterp_column(s,n,k=5):

    y = s.reindex(list(range(n-k,n))+list(range(n+1,n+1+k))) #取数

    y = y[y.notnull()] #剔除空值
    # print(y);exit();
    return lagrange(y.index,list(y))(n) #插值并返回插值结果

#逐个元素判断是否需要插值
for i in data.columns:
    for j in range(len(data)):
        # print(data)
        # exit()
        if(data[i].isnull())[j]: #如果为空就插值   data[i].isnull()返回true/false结果矩阵
            # print(data[i].isnull());exit();                         #
            data[i][j] = ployinterp_column(data[i],j)

data.to_excel(outputfile,header=None,index=False)

猜你喜欢

转载自www.cnblogs.com/hanshuai0921/p/9179221.html