时间序列一维插值

15195775117

1、需求缘起

一维序列插值是常用功能，我常用scipy.interpolate.interp1d来实现，
但是，这个函数对时间格式的序列用不了，
于是，我想着，把时间转为浮点型，再插值不就行了！

2、源代码

import numpy as np
from datetime import datetime,timedelta
from scipy.interpolate import interp1d

# 根据时间序列t，及其对应的值序列value，插值出时间序列t2的值序列
def time_interp1d(t,value,t2):

    # np.datetime64序列转浮点型序列：
    float_t=t.astype("float")
    float_t2=t2.astype("float")

    # 拟合函数:
    f=interp1d(float_t,value)
    value2=f(float_t2)
    return value2

if __name__=="__main__":

    import matplotlib.pyplot as plt

    t0=datetime(2022,9,7,0)
    t1=datetime(2022,9,7,12)

    # 已知的时间序列与值：
    t=np.arange(t0,t1,timedelta(minutes=37))
    value=np.random.randn( len(t) )

    # 想知道以下时间序列的值：
    t2=np.arange(t0,t1,timedelta(hours=1))

    value2=time_interp1d(t,value,t2)

    plt.plot(t,value,c='blue',label='raw data')
    plt.scatter(t2,value2,c='red',label='interp data')
    plt.legend()
    plt.show()

xtzhou1981

除年份外，时间转换成序号1,2,3~n这样作为自变量是比较常用的做法。年份的话就以年份为自变量。