直接上代码,需要的拿去结合自己修改:
import numpy as np
from scipy import interpolate
import matplotlib.pyplot as plt
import time, datetime
from datetime import datetime, date, timedelta
x = ['2021-5-10', '2021-5-11', '2021-5-12', '2021-5-13', '2021-5-16']
y = [3.4783, 1.25, 1.1111, 1.1111, 2.26]
plt.scatter(x, y)
plt.plot(x, y)
plt.show()
我们可以看到以上demo日期是断开的,我们需要获取断开的日期,也就是14、15号
insert_x = []
lxs = len(x)
if lxs < 10:
# 判断是否是连续日期,并对不连续的日期进行时间插值
for i in range(len(x)):
if i+1 == len(x):
break
t1 = int(time.mktime(time.strptime(x[i],"%Y-%m-%d")))
t2 = int(time.mktime(time.strptime(x[i+1],"%Y-%m-%d")))
differ = (datetime.fromtimestamp(t2)-datetime.fromtimestamp(t1)).days
# print("相差",differ,"天")
while differ != 1:
differ -= 1
tmp = (datetime.fromtimestamp(t2) + timedelta(days = -differ)).strftime("%Y-%m-%d")
insert_x.append(tmp)
print(insert_x)