使用matplotlib的X_train,X_test,Y_pred折线图

时间:2019-05-02 00:01:23

标签: python matplotlib data-science

我正在做一个项目股票市场预测,但是我坚持将所有X_train,X_test和Y_pred放在一个像这样的figure 1

我是在使用randomforestregressor算法之后进行的

我的df.head()看起来像这样

    timestamp   open    high    low         close   volume
0   2019-04-12  199.20  200.14  196.2100    198.87  27736755
1   2019-04-11  200.85  201.00  198.4431    198.95  20900808
2   2019-04-10  198.68  200.74  198.1800    200.62  21695288
3   2019-04-09  200.32  202.85  199.2300    199.50  35768237
4   2019-04-08  196.42  200.23  196.3400    200.10  25881697 

这是我要获取类似figure 1这样的输出的方法,但是我在图中没有得到任何东西。

代码:

plt.plot(x=df['timestamp'], y = df['close'], style = 'b-', grid = True)
plt.plot(x=df['timestamp'], y = df['close'], style = 'r-', grid = True)
plt.plot(x=df['timestamp'], y = df['close'], style = 'g-', grid = True)
plt.legend(['X_train', 'X_test', 'Y_pred'])
plt.xlabel("date")
plt.ylabel("USD")

输出

Text(0,0.5,'USD')

figure 2

任何帮助将不胜感激!谢谢。

0 个答案:

没有答案