結果 : how to plot train and test data in python