train_X = []
train_y = []
# time_step = n 表示用n天的数据做预测
time_step = 4
# time_pred = 0 表示直接预测下一天的数据
time_move = 0
a = [i for i in range(10)]
for i in range(len(a) - time_step - time_move):
train_X.append([a[i:i + time_step]])
train_y.append([a[i + time_step + time_move]])
print(train_X)
print(train_y)