■ trainPredictPlot[look_back:len(TrainPredict)+look_back, :] = TrainPredict
훈련 데이터를 예측한 결괏값을 배열에 넣는 코드입니다. 훈련 데이터를 예측한 결괏값의 첫 번째 값은 1일차의 확진자 수가 아니라 4일차의 확진자 수입니다. 1, 2, 3일차의 확진자 수를 보고 4일차를 예측했기 때문입니다. 그러므로 처음 값(look_back)을 건너뛴 다음 그래프를 그려야 합니다. 이를 위해 훈련 데이터를 예측한 결괏값의 구간을 다음과 같이 정해 줍니다(trainPredictPlot[look_back:len(TrainPredict)+look_back, :]).
그림 17-21 | TrainPredict 배열 삽입
그림과 같이 처음 3일치는 건너뛰고 4일차(look_back)부터 TrainPredict 배열의 원소 수까지(len(TrainPredict)+look_back)를 범위로 지정해 줍니다.