代码：data_ori是整个单变量一维数据集合，先构造出形状，使得train与test要画的数据结构斗鱼data_ori一致，再往里面填nan值，最后再把有效数据填入。
import matplotlib.pyplot as plt import pandas as pd split_ratio = 100 test_num = 140 #取140个数据做测试 data_ori = selected_dataset[0:test_num] train_data = data_ori.iloc[0:split_ratio]#除了最后一列不要,代码还可以更加完善可读性高 test_data = data_ori.iloc[split_ratio+20:test_num]#这个20是为了演示下方图蓝色部分 trainPredictPlot = numpy.empty_like(data_ori) trainPredictPlot[:] = numpy.nan trainPredictPlot[:split_ratio] = train_data testPredictPlot = numpy.empty_like(data_ori) testPredictPlot[:] = numpy.nan testPredictPlot[split_ratio+20:test_num] = test_data plt.plot(data_ori)#全部 plt.plot(trainPredictPlot)#训练预测 plt.plot(testPredictPlot)#测试预测 plt.show()
data_ori图
Train_predict与test_predict图
data_ori与Train_predict与test_predict图 ，第一段为train_predict,第二段为data_ori，第三段为test_predict
